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

Public Member Functions

 ModelFaceBoxes (const std::string &modelFileName, float confidenceThreshold, bool useAutoResize, float boxIOUThreshold, const std::string &layout="")
 
std::unique_ptr< ResultBasepostprocess (InferenceResult &infResult) override
 
- Public Member Functions inherited from DetectionModel
 DetectionModel (const std::string &modelFileName, float confidenceThreshold, bool useAutoResize, const std::vector< std::string > &labels, const std::string &layout="")
 
- Public Member Functions inherited from ImageModel
 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)
 
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)
 

Static Public Attributes

static const int INIT_VECTOR_SIZE = 200
 

Protected Member Functions

void prepareInputsOutputs (std::shared_ptr< ov::Model > &model) override
 
void priorBoxes (const std::vector< std::pair< size_t, size_t >> &featureMaps)
 
- Protected Member Functions inherited from DetectionModel
std::string getLabelName (size_t labelID)
 
- Protected Member Functions inherited from ModelBase
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

size_t maxProposalsCount
 
const float boxIOUThreshold
 
const std::vector< float > variance
 
const std::vector< int > steps
 
const std::vector< std::vector< int > > minSizes
 
std::vector< Anchoranchors
 
- Protected Attributes inherited from DetectionModel
float confidenceThreshold
 
std::vector< std::string > labels
 
- Protected Attributes inherited from ImageModel
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

- Static Public Member Functions inherited from DetectionModel
static std::vector< std::string > loadLabels (const std::string &labelFilename)
 

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