plugin  0.1.0
Public Member Functions | Protected Attributes | List of all members
ImageModel Class Reference
Inheritance diagram for ImageModel:
Inheritance graph
[legend]
Collaboration diagram for ImageModel:
Collaboration graph
[legend]

Public Member Functions

 ImageModel (const std::string &modelFileName, bool useAutoResize, const std::string &layout="")
 
std::shared_ptr< InternalModelDatapreprocess (const InputData &inputData, ov::InferRequest &request) override
 
- Public Member Functions inherited from ModelBase
 ModelBase (const std::string &modelFileName, const std::string &layout="")
 
virtual ov::CompiledModel compileModel (const ModelConfig &config, ov::Core &core)
 
virtual void onLoadCompleted (const std::vector< ov::InferRequest > &requests)
 
virtual std::unique_ptr< ResultBasepostprocess (InferenceResult &infResult)=0
 
const std::vector< std::string > & getOutputsNames () const
 
const std::vector< std::string > & getInputsNames () const
 
std::string getModelFileName ()
 
void setInputsPreprocessing (bool reverseInputChannels, const std::string &meanValues, const std::string &scaleValues)
 

Protected Attributes

bool useAutoResize
 
size_t netInputHeight = 0
 
size_t netInputWidth = 0
 
cv::InterpolationFlags interpolationMode = cv::INTER_LINEAR
 
RESIZE_MODE resizeMode = RESIZE_FILL
 
- Protected Attributes inherited from ModelBase
InputTransform inputTransform = InputTransform()
 
std::vector< std::string > inputsNames
 
std::vector< std::string > outputsNames
 
ov::CompiledModel compiledModel
 
std::string modelFileName
 
ModelConfig config = {}
 
std::map< std::string, ov::Layout > inputsLayouts
 

Additional Inherited Members

- Protected Member Functions inherited from ModelBase
virtual void prepareInputsOutputs (std::shared_ptr< ov::Model > &model)=0
 
virtual void setBatch (std::shared_ptr< ov::Model > &model)
 
std::shared_ptr< ov::Model > prepareModel (ov::Core &core)
 
ov::Layout getInputLayout (const ov::Output< ov::Node > &input)
 

Constructor & Destructor Documentation

◆ ImageModel()

ImageModel::ImageModel ( const std::string &  modelFileName,
bool  useAutoResize,
const std::string &  layout = "" 
)

Constructor

Parameters
modelFileNamename of model to load
useAutoResize- if true, image is resized by openvino
layout- model input layout

The documentation for this class was generated from the following files: