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 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 105 of file nvdsinfer_model_builder.h.

Member Function Documentation

getModelName()

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

inlineoverride

Definition at line 112 of file nvdsinfer_model_builder.h.

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

hasFullDimsSupported()

bool nvdsinfer::CustomModelParser::hasFullDimsSupported ( ) const

inlineoverride

Definition at line 118 of file nvdsinfer_model_builder.h.

isValid()

bool nvdsinfer::CustomModelParser::isValid ( ) const

inlineoverridevirtual

Implements nvdsinfer::BaseModelParser.

Definition at line 107 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