NVIDIA DeepStream SDK API Reference

8.0 Release
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 sources/libs/nvdsinfer/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
 
 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< DlLibHandlem_LibHandle
 

Constructor & Destructor Documentation

◆ OnnxModelParser() [1/2]

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

◆ ~OnnxModelParser() [1/2]

nvdsinfer::OnnxModelParser::~OnnxModelParser ( )
overridedefault

◆ OnnxModelParser() [2/2]

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

◆ ~OnnxModelParser() [2/2]

nvdsinfer::OnnxModelParser::~OnnxModelParser ( )
overridedefault

Member Function Documentation

◆ getModelName() [1/2]

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

◆ getModelName() [2/2]

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

◆ hasFullDimsSupported() [1/2]

bool nvdsinfer::OnnxModelParser::hasFullDimsSupported ( ) const
inlineoverride

◆ hasFullDimsSupported() [2/2]

bool nvdsinfer::OnnxModelParser::hasFullDimsSupported ( ) const
inlineoverride

◆ isValid() [1/2]

bool nvdsinfer::OnnxModelParser::isValid ( ) const
inlineoverridevirtual

◆ isValid() [2/2]

bool nvdsinfer::OnnxModelParser::isValid ( ) const
inlineoverridevirtual

◆ parseModel() [1/2]

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

◆ parseModel() [2/2]

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

Field Documentation

◆ m_LibHandle

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

◆ m_ModelParams

NvDsInferContextInitParams nvdsinfer::BaseModelParser::m_ModelParams
protectedinherited

◆ m_OnnxParser

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

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