NVIDIA DeepStream SDK API Reference
6.4 Release
Public Member Functions | Protected Member Functions | Protected Attributes
nvdsinfer::ClassifyPostprocessor Class Reference

Detailed Description

Implementation of post-processing class for classification networks.

Definition at line 314 of file nvdsinfer_context_impl.h.

Inheritance diagram for nvdsinfer::ClassifyPostprocessor:
Inheritance graph
[legend]
Collaboration diagram for nvdsinfer::ClassifyPostprocessor:
Collaboration graph
[legend]

Public Member Functions

 ClassifyPostprocessor (int id, int gpuId=0)
 
NvDsInferStatus initResource (const NvDsInferContextInitParams &initParams) override
 
void setDlHandle (const std::shared_ptr< DlLibHandle > &dlHandle)
 
void setNetworkInfo (const NvDsInferNetworkInfo &info)
 
void setAllLayerInfo (std::vector< NvDsInferBatchDimsLayerInfo > &info)
 
void setOutputLayerInfo (std::vector< NvDsInferBatchDimsLayerInfo > &info)
 
void setLoggingFunc (const NvDsInferLoggingFunc &func)
 
const std::vector< std::vector< std::string > > & getLabels () const
 
bool needInputCopy () const
 
bool needOutputCopyB4Processing () const
 
virtual NvDsInferStatus copyBuffersToHostMemory (NvDsInferBatch &buffer, CudaStream &mainStream)
 
virtual NvDsInferStatus postProcessHost (NvDsInferBatch &buffer, NvDsInferContextBatchOutput &output)
 
void freeBatchOutput (NvDsInferContextBatchOutput &batchOutput)
 

Protected Member Functions

NvDsInferStatus parseLabelsFile (const std::string &path)
 
NvDsInferStatus allocDeviceResource ()
 
void releaseFrameOutput (NvDsInferFrameOutput &frameOutput)
 

Protected Attributes

NvDsInferNetworkType m_NetworkType = NvDsInferNetworkType_Other
 
int m_UniqueID = 0
 
uint32_t m_GpuID = 0
 
NvDsInferLoggingFunc m_LoggingFunc
 
std::shared_ptr< DlLibHandle > m_CustomLibHandle
 
bool m_CopyInputToHostBuffers = false
 
bool m_disableOutputHostCopy = false
 
bool m_DumpOpTensor = false
 
std::vector< std::pair< std::string, std::string > > m_DumpOpTensorFiles
 
bool m_OverwriteOpTensor = false
 
std::vector< std::pair< std::string, int > > m_OverwriteOpTensorFilePairs
 
std::vector< std::ifstream * > m_OverwriteOpTensorFiles
 
NvDsInferNetworkInfo m_NetworkInfo = {0}
 
std::vector< NvDsInferLayerInfo > m_AllLayerInfo
 
std::vector< NvDsInferLayerInfo > m_OutputLayerInfo
 
std::vector< std::vector< std::string > > m_Labels
 

Constructor & Destructor Documentation

◆ ClassifyPostprocessor()

nvdsinfer::ClassifyPostprocessor::ClassifyPostprocessor ( int  id,
int  gpuId = 0 
)
inline

Definition at line 317 of file nvdsinfer_context_impl.h.

Member Function Documentation

◆ allocDeviceResource()

NvDsInferStatus nvdsinfer::InferPostprocessor::allocDeviceResource ( )
protectedinherited

◆ copyBuffersToHostMemory()

virtual NvDsInferStatus nvdsinfer::InferPostprocessor::copyBuffersToHostMemory ( NvDsInferBatch &  buffer,
CudaStream &  mainStream 
)
virtualinherited

◆ freeBatchOutput()

void nvdsinfer::InferPostprocessor::freeBatchOutput ( NvDsInferContextBatchOutput &  batchOutput)
inherited

◆ getLabels()

const std::vector<std::vector<std::string> >& nvdsinfer::InferPostprocessor::getLabels ( ) const
inlineinherited

Definition at line 146 of file nvdsinfer_context_impl.h.

References nvdsinfer::InferPostprocessor::m_Labels.

◆ initResource()

NvDsInferStatus nvdsinfer::ClassifyPostprocessor::initResource ( const NvDsInferContextInitParams &  initParams)
overridevirtual

Reimplemented from nvdsinfer::InferPostprocessor.

◆ needInputCopy()

bool nvdsinfer::InferPostprocessor::needInputCopy ( ) const
inlineinherited

Definition at line 150 of file nvdsinfer_context_impl.h.

References nvdsinfer::InferPostprocessor::m_CopyInputToHostBuffers.

◆ needOutputCopyB4Processing()

bool nvdsinfer::InferPostprocessor::needOutputCopyB4Processing ( ) const
inlineinherited

Definition at line 152 of file nvdsinfer_context_impl.h.

References nvdsinfer::InferPostprocessor::m_disableOutputHostCopy.

◆ parseLabelsFile()

NvDsInferStatus nvdsinfer::InferPostprocessor::parseLabelsFile ( const std::string &  path)
protectedinherited

◆ postProcessHost()

virtual NvDsInferStatus nvdsinfer::InferPostprocessor::postProcessHost ( NvDsInferBatch &  buffer,
NvDsInferContextBatchOutput &  output 
)
virtualinherited

◆ releaseFrameOutput()

void nvdsinfer::InferPostprocessor::releaseFrameOutput ( NvDsInferFrameOutput &  frameOutput)
protectedinherited

◆ setAllLayerInfo()

void nvdsinfer::InferPostprocessor::setAllLayerInfo ( std::vector< NvDsInferBatchDimsLayerInfo > &  info)
inlineinherited

Definition at line 132 of file nvdsinfer_context_impl.h.

References nvdsinfer::InferPostprocessor::m_AllLayerInfo.

◆ setDlHandle()

void nvdsinfer::InferPostprocessor::setDlHandle ( const std::shared_ptr< DlLibHandle > &  dlHandle)
inlineinherited

Definition at line 124 of file nvdsinfer_context_impl.h.

References nvdsinfer::InferPostprocessor::m_CustomLibHandle.

◆ setLoggingFunc()

void nvdsinfer::InferPostprocessor::setLoggingFunc ( const NvDsInferLoggingFunc &  func)
inlineinherited

Definition at line 142 of file nvdsinfer_context_impl.h.

References nvdsinfer::InferPostprocessor::m_LoggingFunc.

◆ setNetworkInfo()

void nvdsinfer::InferPostprocessor::setNetworkInfo ( const NvDsInferNetworkInfo &  info)
inlineinherited

Definition at line 128 of file nvdsinfer_context_impl.h.

References nvdsinfer::InferPostprocessor::m_NetworkInfo.

◆ setOutputLayerInfo()

void nvdsinfer::InferPostprocessor::setOutputLayerInfo ( std::vector< NvDsInferBatchDimsLayerInfo > &  info)
inlineinherited

Definition at line 137 of file nvdsinfer_context_impl.h.

References nvdsinfer::InferPostprocessor::m_OutputLayerInfo.

Field Documentation

◆ m_AllLayerInfo

std::vector<NvDsInferLayerInfo> nvdsinfer::InferPostprocessor::m_AllLayerInfo
protectedinherited

Definition at line 199 of file nvdsinfer_context_impl.h.

Referenced by nvdsinfer::InferPostprocessor::setAllLayerInfo().

◆ m_CopyInputToHostBuffers

bool nvdsinfer::InferPostprocessor::m_CopyInputToHostBuffers = false
protectedinherited

Definition at line 190 of file nvdsinfer_context_impl.h.

Referenced by nvdsinfer::InferPostprocessor::needInputCopy().

◆ m_CustomLibHandle

std::shared_ptr<DlLibHandle> nvdsinfer::InferPostprocessor::m_CustomLibHandle
protectedinherited

Definition at line 189 of file nvdsinfer_context_impl.h.

Referenced by nvdsinfer::InferPostprocessor::setDlHandle().

◆ m_disableOutputHostCopy

bool nvdsinfer::InferPostprocessor::m_disableOutputHostCopy = false
protectedinherited

Definition at line 191 of file nvdsinfer_context_impl.h.

Referenced by nvdsinfer::InferPostprocessor::needOutputCopyB4Processing().

◆ m_DumpOpTensor

bool nvdsinfer::InferPostprocessor::m_DumpOpTensor = false
protectedinherited

Definition at line 192 of file nvdsinfer_context_impl.h.

◆ m_DumpOpTensorFiles

std::vector<std::pair<std::string, std::string> > nvdsinfer::InferPostprocessor::m_DumpOpTensorFiles
protectedinherited

Definition at line 193 of file nvdsinfer_context_impl.h.

◆ m_GpuID

uint32_t nvdsinfer::InferPostprocessor::m_GpuID = 0
protectedinherited

Definition at line 185 of file nvdsinfer_context_impl.h.

◆ m_Labels

std::vector<std::vector<std::string> > nvdsinfer::InferPostprocessor::m_Labels
protectedinherited

Definition at line 203 of file nvdsinfer_context_impl.h.

Referenced by nvdsinfer::InferPostprocessor::getLabels().

◆ m_LoggingFunc

NvDsInferLoggingFunc nvdsinfer::InferPostprocessor::m_LoggingFunc
protectedinherited

Definition at line 186 of file nvdsinfer_context_impl.h.

Referenced by nvdsinfer::InferPostprocessor::setLoggingFunc().

◆ m_NetworkInfo

NvDsInferNetworkInfo nvdsinfer::InferPostprocessor::m_NetworkInfo = {0}
protectedinherited

Definition at line 198 of file nvdsinfer_context_impl.h.

Referenced by nvdsinfer::InferPostprocessor::setNetworkInfo().

◆ m_NetworkType

NvDsInferNetworkType nvdsinfer::InferPostprocessor::m_NetworkType = NvDsInferNetworkType_Other
protectedinherited

Definition at line 182 of file nvdsinfer_context_impl.h.

◆ m_OutputLayerInfo

std::vector<NvDsInferLayerInfo> nvdsinfer::InferPostprocessor::m_OutputLayerInfo
protectedinherited

Definition at line 200 of file nvdsinfer_context_impl.h.

Referenced by nvdsinfer::InferPostprocessor::setOutputLayerInfo().

◆ m_OverwriteOpTensor

bool nvdsinfer::InferPostprocessor::m_OverwriteOpTensor = false
protectedinherited

Definition at line 194 of file nvdsinfer_context_impl.h.

◆ m_OverwriteOpTensorFilePairs

std::vector<std::pair<std::string, int> > nvdsinfer::InferPostprocessor::m_OverwriteOpTensorFilePairs
protectedinherited

Definition at line 195 of file nvdsinfer_context_impl.h.

◆ m_OverwriteOpTensorFiles

std::vector<std::ifstream *> nvdsinfer::InferPostprocessor::m_OverwriteOpTensorFiles
protectedinherited

Definition at line 196 of file nvdsinfer_context_impl.h.

◆ m_UniqueID

int nvdsinfer::InferPostprocessor::m_UniqueID = 0
protectedinherited

Definition at line 184 of file nvdsinfer_context_impl.h.

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