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

Public Member Functions

 ModelBase (const std::string &modelFileName, const std::string &layout="")
 
virtual std::shared_ptr< InternalModelDatapreprocess (const InputData &inputData, ov::InferRequest &request)=0
 
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 Member Functions

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)
 

Protected Attributes

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
 

Member Function Documentation

◆ prepareModel()

std::shared_ptr< ov::Model > ModelBase::prepareModel ( ov::Core &  core)
protected

Read model


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