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 160 of file 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

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

Constructor & Destructor Documentation

CustomModelParser()

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

~CustomModelParser()

nvdsinfer::CustomModelParser::~CustomModelParser ( )

inline

Definition at line 166 of file nvdsinfer_model_builder.h.

Member Function Documentation

getModelName()

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

inlineoverride

Definition at line 173 of file nvdsinfer_model_builder.h.

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

hasFullDimsSupported()

bool nvdsinfer::CustomModelParser::hasFullDimsSupported ( ) const

inlineoverride

Definition at line 179 of file nvdsinfer_model_builder.h.

isValid()

bool nvdsinfer::CustomModelParser::isValid ( ) const

inlineoverridevirtual

Implements nvdsinfer::BaseModelParser.

Definition at line 168 of file nvdsinfer_model_builder.h.

Referenced by getModelName().

parseModel()

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 nvdsinfer_model_builder.h.

m_ModelParams

NvDsInferContextInitParams nvdsinfer::BaseModelParser::m_ModelParams

protectedinherited

Definition at line 64 of file nvdsinfer_model_builder.h.

---

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

• nvdsinfer_model_builder.h