nvdsinfer::OnnxModelParser Class Reference

Detailed Description

Implementation of ModelParser for ONNX models derived from BaseModelParser.

Manages resources internally required for parsing ONNX models.

Definition at line 134 of file nvdsinfer_model_builder.h.

Inheritance diagram for nvdsinfer::OnnxModelParser:

Collaboration diagram for nvdsinfer::OnnxModelParser:

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

Protected Attributes
UniquePtrWDestroy< nvonnxparser::IParser >	m_OnnxParser
NvDsInferContextInitParams	m_ModelParams
std::shared_ptr< DlLibHandle >	m_LibHandle

Constructor & Destructor Documentation

OnnxModelParser()

nvdsinfer::OnnxModelParser::OnnxModelParser ( const NvDsInferContextInitParams &  initParams, const std::shared_ptr< DlLibHandle > &  handle = nullptr  )

inline

Definition at line 137 of file nvdsinfer_model_builder.h.

~OnnxModelParser()

nvdsinfer::OnnxModelParser::~OnnxModelParser ( )

overridedefault

Member Function Documentation

getModelName()

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

inlineoverride

Definition at line 143 of file nvdsinfer_model_builder.h.

hasFullDimsSupported()

bool nvdsinfer::OnnxModelParser::hasFullDimsSupported ( ) const

inlineoverride

Definition at line 145 of file nvdsinfer_model_builder.h.

isValid()

bool nvdsinfer::OnnxModelParser::isValid ( ) const

inlineoverridevirtual

Implements nvdsinfer::BaseModelParser.

Definition at line 142 of file nvdsinfer_model_builder.h.

parseModel()

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

override

Field Documentation

m_LibHandle

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

protectedinherited

Definition at line 66 of file nvdsinfer_model_builder.h.

m_ModelParams

NvDsInferContextInitParams nvdsinfer::BaseModelParser::m_ModelParams

protectedinherited

Definition at line 65 of file nvdsinfer_model_builder.h.

m_OnnxParser

UniquePtrWDestroy<nvonnxparser::IParser> nvdsinfer::OnnxModelParser::m_OnnxParser

protected

Definition at line 151 of file nvdsinfer_model_builder.h.

---

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

• nvdsinfer_model_builder.h