nvdsinfer::CustomModelParser Class Reference

Detailed Description

Implementation of ModelParser for custom models.

This implementation will look for the function symbol "NvDsInferCreateModelParser" in the custom library handle passed to it. It will call the NvDsInferCreateModelParser to get an instance of the IModelParser implementation required to parse the user's custom model.

Definition at line 99 of file sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

Inheritance diagram for nvdsinfer::CustomModelParser:

Collaboration diagram for nvdsinfer::CustomModelParser:

Public Member Functions
	CustomModelParser (const NvDsInferContextInitParams &initParams, const std::shared_ptr< DlLibHandle > &handle)
	~CustomModelParser ()
bool	isValid () const override
const char *	getModelName () const override
NvDsInferStatus	parseModel (nvinfer1::INetworkDefinition &network) override
bool	hasFullDimsSupported () const override
	CustomModelParser (const NvDsInferContextInitParams &initParams, const std::shared_ptr< DlLibHandle > &handle)
	~CustomModelParser ()
bool	isValid () const override
const char *	getModelName () const override
NvDsInferStatus	parseModel (nvinfer1::INetworkDefinition &network) override
bool	hasFullDimsSupported () const override

Protected Attributes
NvDsInferContextInitParams	m_ModelParams
std::shared_ptr< DlLibHandle >	m_LibHandle

Constructor & Destructor Documentation

CustomModelParser() [1/2]

nvdsinfer::CustomModelParser::CustomModelParser	(	const NvDsInferContextInitParams &	initParams,
		const std::shared_ptr< DlLibHandle > &	handle
	)

~CustomModelParser() [1/2]

nvdsinfer::CustomModelParser::~CustomModelParser ( )

inline

Definition at line 105 of file sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

CustomModelParser() [2/2]

nvdsinfer::CustomModelParser::CustomModelParser	(	const NvDsInferContextInitParams &	initParams,
		const std::shared_ptr< DlLibHandle > &	handle
	)

~CustomModelParser() [2/2]

nvdsinfer::CustomModelParser::~CustomModelParser ( )

inline

Definition at line 105 of file 9.0/sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

Member Function Documentation

getModelName() [1/2]

const char* nvdsinfer::CustomModelParser::getModelName ( ) const

inlineoverride

Definition at line 112 of file sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

References isValid(), and nvdsinfer::safeStr().

getModelName() [2/2]

const char* nvdsinfer::CustomModelParser::getModelName ( ) const

inlineoverride

Definition at line 112 of file 9.0/sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

References isValid(), and nvdsinfer::safeStr().

hasFullDimsSupported() [1/2]

bool nvdsinfer::CustomModelParser::hasFullDimsSupported ( ) const

inlineoverride

Definition at line 118 of file sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

hasFullDimsSupported() [2/2]

bool nvdsinfer::CustomModelParser::hasFullDimsSupported ( ) const

inlineoverride

Definition at line 118 of file 9.0/sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

isValid() [1/2]

bool nvdsinfer::CustomModelParser::isValid ( ) const

inlineoverridevirtual

Implements nvdsinfer::BaseModelParser.

Definition at line 107 of file 9.0/sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

isValid() [2/2]

bool nvdsinfer::CustomModelParser::isValid ( ) const

inlineoverridevirtual

Implements nvdsinfer::BaseModelParser.

Definition at line 107 of file sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

Referenced by getModelName().

parseModel() [1/2]

NvDsInferStatus nvdsinfer::CustomModelParser::parseModel ( nvinfer1::INetworkDefinition &  network )

override

parseModel() [2/2]

NvDsInferStatus nvdsinfer::CustomModelParser::parseModel ( nvinfer1::INetworkDefinition &  network )

override

Field Documentation

m_LibHandle

std::shared_ptr< DlLibHandle > nvdsinfer::BaseModelParser::m_LibHandle

protectedinherited

Definition at line 65 of file sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

m_ModelParams

NvDsInferContextInitParams nvdsinfer::BaseModelParser::m_ModelParams

protectedinherited

Definition at line 64 of file sources/libs/nvdsinfer/nvdsinfer_model_builder.h.

---

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

• sources/libs/nvdsinfer/nvdsinfer_model_builder.h