Class InferBase
Defined in File infer.hpp
Derived Types
public holoscan::inference::OnnxInfer
(Class OnnxInfer)public holoscan::inference::TrtInfer
(Class TrtInfer)
-
class InferBase
Base Inference Class.
Subclassed by holoscan::inference::OnnxInfer, holoscan::inference::TrtInfer
Public Functions
Does the Core inference.
- Parameters
input_data – Input DataBuffer
output_buffer – Output DataBuffer, is populated with inferred results
- Returns
InferStatus
-
inline virtual std::vector<int64_t> get_input_dims() const
Get input data dimensions to the model.
- Returns
Vector of values as dimension
-
inline virtual std::vector<int64_t> get_output_dims() const
Get output data dimensions from the model.
- Returns
Vector of values as dimension
- inline virtual void cleanup()