NVIDIA DeepStream SDK API Reference
6.4 Release
Public Member Functions | Protected Member Functions | Protected Attributes
ClassifyModelPostProcessor Class Reference

Detailed Description

Definition at line 37 of file post_processor_classify.h.

Inheritance diagram for ClassifyModelPostProcessor:
Inheritance graph
[legend]
Collaboration diagram for ClassifyModelPostProcessor:
Collaboration graph
[legend]

Public Member Functions

 ClassifyModelPostProcessor (int id, int gpuId=0)
 
 ~ClassifyModelPostProcessor () override=default
 
NvDsPostProcessStatus initResource (NvDsPostProcessContextInitParams &initParams) override
 
NvDsPostProcessStatus parseEachFrame (const std::vector< NvDsInferLayerInfo > &outputLayers, NvDsPostProcessFrameOutput &result) override
 
void attachMetadata (NvBufSurface *surf, gint batch_idx, NvDsBatchMeta *batch_meta, NvDsFrameMeta *frame_meta, NvDsObjectMeta *object_meta, NvDsObjectMeta *parent_obj_meta, NvDsPostProcessFrameOutput &detection_output, NvDsPostProcessDetectionParams *all_params, std::set< gint > &filterOutClassIds, int32_t unique_id, gboolean output_instance_mask, gboolean process_full_frame, float segmentationThreshold, gboolean maintain_aspect_ratio) override
 
void mergeClassificationOutput (NvDsPostProcessObjectHistory &history, NvDsPostProcessObjectInfo &new_result)
 
void releaseFrameOutput (NvDsPostProcessFrameOutput &frameOutput) override
 
const std::vector< std::vector< std::string > > & getLabels () const
 
void freeBatchOutput (NvDsPostProcessBatchOutput &batchOutput)
 
void setNetworkInfo (NvDsInferNetworkInfo networkInfo)
 

Protected Member Functions

NvDsPostProcessStatus parseLabelsFile (const std::string &path)
 

Protected Attributes

NvDsPostProcessNetworkType m_NetworkType = NvDsPostProcessNetworkType_Other
 
int m_UniqueID = 0
 
uint32_t m_GpuID = 0
 
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

◆ ClassifyModelPostProcessor()

ClassifyModelPostProcessor::ClassifyModelPostProcessor ( int  id,
int  gpuId = 0 
)
inline

Definition at line 40 of file post_processor_classify.h.

◆ ~ClassifyModelPostProcessor()

ClassifyModelPostProcessor::~ClassifyModelPostProcessor ( )
overridedefault

Member Function Documentation

◆ attachMetadata()

void ClassifyModelPostProcessor::attachMetadata ( NvBufSurface *  surf,
gint  batch_idx,
NvDsBatchMeta *  batch_meta,
NvDsFrameMeta *  frame_meta,
NvDsObjectMeta *  object_meta,
NvDsObjectMeta *  parent_obj_meta,
NvDsPostProcessFrameOutput &  detection_output,
NvDsPostProcessDetectionParams *  all_params,
std::set< gint > &  filterOutClassIds,
int32_t  unique_id,
gboolean  output_instance_mask,
gboolean  process_full_frame,
float  segmentationThreshold,
gboolean  maintain_aspect_ratio 
)
overridevirtual

Implements ModelPostProcessor.

◆ freeBatchOutput()

void ModelPostProcessor::freeBatchOutput ( NvDsPostProcessBatchOutput &  batchOutput)
inherited

◆ getLabels()

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

Definition at line 69 of file post_processor.h.

References ModelPostProcessor::m_Labels.

◆ initResource()

NvDsPostProcessStatus ClassifyModelPostProcessor::initResource ( NvDsPostProcessContextInitParams &  initParams)
overridevirtual

Reimplemented from ModelPostProcessor.

◆ mergeClassificationOutput()

void ClassifyModelPostProcessor::mergeClassificationOutput ( NvDsPostProcessObjectHistory &  history,
NvDsPostProcessObjectInfo &  new_result 
)

◆ parseEachFrame()

NvDsPostProcessStatus ClassifyModelPostProcessor::parseEachFrame ( const std::vector< NvDsInferLayerInfo > &  outputLayers,
NvDsPostProcessFrameOutput &  result 
)
overridevirtual

Implements ModelPostProcessor.

◆ parseLabelsFile()

NvDsPostProcessStatus ModelPostProcessor::parseLabelsFile ( const std::string &  path)
protectedinherited

◆ releaseFrameOutput()

void ClassifyModelPostProcessor::releaseFrameOutput ( NvDsPostProcessFrameOutput &  frameOutput)
overridevirtual

Implements ModelPostProcessor.

◆ setNetworkInfo()

void ModelPostProcessor::setNetworkInfo ( NvDsInferNetworkInfo  networkInfo)
inlineinherited

Definition at line 74 of file post_processor.h.

References ModelPostProcessor::m_NetworkInfo.

Field Documentation

◆ m_AllLayerInfo

std::vector<NvDsInferLayerInfo> ModelPostProcessor::m_AllLayerInfo
protectedinherited

Definition at line 116 of file post_processor.h.

◆ m_GpuID

uint32_t ModelPostProcessor::m_GpuID = 0
protectedinherited

Definition at line 112 of file post_processor.h.

◆ m_Labels

std::vector<std::vector<std::string> > ModelPostProcessor::m_Labels
protectedinherited

Definition at line 120 of file post_processor.h.

Referenced by ModelPostProcessor::getLabels().

◆ m_NetworkInfo

NvDsInferNetworkInfo ModelPostProcessor::m_NetworkInfo = {0}
protectedinherited

Definition at line 115 of file post_processor.h.

Referenced by ModelPostProcessor::setNetworkInfo().

◆ m_NetworkType

NvDsPostProcessNetworkType ModelPostProcessor::m_NetworkType = NvDsPostProcessNetworkType_Other
protectedinherited

Definition at line 109 of file post_processor.h.

◆ m_OutputLayerInfo

std::vector<NvDsInferLayerInfo> ModelPostProcessor::m_OutputLayerInfo
protectedinherited

Definition at line 117 of file post_processor.h.

◆ m_UniqueID

int ModelPostProcessor::m_UniqueID = 0
protectedinherited

Definition at line 111 of file post_processor.h.

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