Yolo Class Reference

Detailed Description

Definition at line 65 of file yolo.h.

Inheritance diagram for Yolo:

Collaboration diagram for Yolo:

Public Member Functions
	Yolo (const NetworkInfo &networkInfo)
	~Yolo () override
bool	hasFullDimsSupported () const override
const char *	getModelName () const override
NvDsInferStatus	parseModel (nvinfer1::INetworkDefinition &network) override
nvinfer1::ICudaEngine *	createEngine (nvinfer1::IBuilder *builder, nvinfer1::IBuilderConfig *config)

Protected Attributes
const std::string	m_NetworkType
const std::string	m_ConfigFilePath
const std::string	m_WtsFilePath
const std::string	m_DeviceType
const std::string	m_InputBlobName
std::vector< TensorInfo >	m_OutputTensors
std::vector< std::map< std::string, std::string > >	m_ConfigBlocks
uint	m_InputH
uint	m_InputW
uint	m_InputC
uint64_t	m_InputSize
std::vector< nvinfer1::Weights >	m_TrtWeights

Constructor & Destructor Documentation

Yolo()

Yolo::Yolo

(

const NetworkInfo &

networkInfo

)

~Yolo()

Yolo::~Yolo ( )

override

Member Function Documentation

createEngine()

nvinfer1::ICudaEngine* Yolo::createEngine	(	nvinfer1::IBuilder *	builder,
		nvinfer1::IBuilderConfig *	config
	)

getModelName()

const char* Yolo::getModelName ( ) const

inlineoverride

Definition at line 70 of file yolo.h.

References m_ConfigFilePath, and m_NetworkType.

hasFullDimsSupported()

bool Yolo::hasFullDimsSupported ( ) const

inlineoverride

Definition at line 69 of file yolo.h.

parseModel()

NvDsInferStatus Yolo::parseModel ( nvinfer1::INetworkDefinition &  network )

override

Field Documentation

m_ConfigBlocks

std::vector<std::map<std::string, std::string> > Yolo::m_ConfigBlocks

protected

Definition at line 86 of file yolo.h.

m_ConfigFilePath

const std::string Yolo::m_ConfigFilePath

protected

Definition at line 81 of file yolo.h.

Referenced by getModelName().

m_DeviceType

const std::string Yolo::m_DeviceType

protected

Definition at line 83 of file yolo.h.

m_InputBlobName

const std::string Yolo::m_InputBlobName

protected

Definition at line 84 of file yolo.h.

m_InputC

uint Yolo::m_InputC

protected

Definition at line 89 of file yolo.h.

m_InputH

uint Yolo::m_InputH

protected

Definition at line 87 of file yolo.h.

m_InputSize

uint64_t Yolo::m_InputSize

protected

Definition at line 90 of file yolo.h.

m_InputW

uint Yolo::m_InputW

protected

Definition at line 88 of file yolo.h.

m_NetworkType

const std::string Yolo::m_NetworkType

protected

Definition at line 80 of file yolo.h.

Referenced by getModelName().

m_OutputTensors

std::vector<TensorInfo> Yolo::m_OutputTensors

protected

Definition at line 85 of file yolo.h.

m_TrtWeights

std::vector<nvinfer1::Weights> Yolo::m_TrtWeights

protected

Definition at line 93 of file yolo.h.

m_WtsFilePath

const std::string Yolo::m_WtsFilePath

protected

Definition at line 82 of file yolo.h.

---

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

• yolo.h