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 72 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
std::unique_ptr< 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 75 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 81 of file nvdsinfer_model_builder.h.

hasFullDimsSupported()

bool nvdsinfer::OnnxModelParser::hasFullDimsSupported ( ) const

inlineoverride

Definition at line 83 of file nvdsinfer_model_builder.h.

isValid()

bool nvdsinfer::OnnxModelParser::isValid ( ) const

inlineoverridevirtual

Implements nvdsinfer::BaseModelParser.

Definition at line 80 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 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.

m_OnnxParser

std::unique_ptr<nvonnxparser::IParser> nvdsinfer::OnnxModelParser::m_OnnxParser

protected

Definition at line 89 of file nvdsinfer_model_builder.h.

---

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

• nvdsinfer_model_builder.h