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

Detailed Description

Definition at line 57 of file post_processor.h.

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

Public Member Functions

virtual ~ModelPostProcessor ()=default
 
virtual NvDsPostProcessStatus initResource (NvDsPostProcessContextInitParams &initParams)
 
const std::vector< std::vector< std::string > > & getLabels () const
 
void freeBatchOutput (NvDsPostProcessBatchOutput &batchOutput)
 
void setNetworkInfo (NvDsInferNetworkInfo networkInfo)
 
virtual NvDsPostProcessStatus parseEachFrame (const std::vector< NvDsInferLayerInfo > &outputLayers, NvDsPostProcessFrameOutput &result)=0
 
virtual 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)=0
 
virtual void releaseFrameOutput (NvDsPostProcessFrameOutput &frameOutput)=0
 

Protected Member Functions

 ModelPostProcessor (NvDsPostProcessNetworkType type, int id, int gpuId)
 
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

◆ ModelPostProcessor()

ModelPostProcessor::ModelPostProcessor ( NvDsPostProcessNetworkType  type,
int  id,
int  gpuId 
)
inlineprotected

Definition at line 61 of file post_processor.h.

◆ ~ModelPostProcessor()

virtual ModelPostProcessor::~ModelPostProcessor ( )
virtualdefault

Member Function Documentation

◆ attachMetadata()

virtual void ModelPostProcessor::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 
)
pure virtual

Implemented in InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, and SegmentationModelPostProcessor.

◆ freeBatchOutput()

void ModelPostProcessor::freeBatchOutput ( NvDsPostProcessBatchOutput &  batchOutput)

◆ getLabels()

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

Definition at line 69 of file post_processor.h.

References m_Labels.

◆ initResource()

virtual NvDsPostProcessStatus ModelPostProcessor::initResource ( NvDsPostProcessContextInitParams &  initParams)
virtual

Reimplemented in InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, and SegmentationModelPostProcessor.

◆ parseEachFrame()

virtual NvDsPostProcessStatus ModelPostProcessor::parseEachFrame ( const std::vector< NvDsInferLayerInfo > &  outputLayers,
NvDsPostProcessFrameOutput &  result 
)
pure virtual

Implemented in InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, and SegmentationModelPostProcessor.

◆ parseLabelsFile()

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

◆ releaseFrameOutput()

virtual void ModelPostProcessor::releaseFrameOutput ( NvDsPostProcessFrameOutput &  frameOutput)
pure virtual

Implemented in InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, and SegmentationModelPostProcessor.

◆ setNetworkInfo()

void ModelPostProcessor::setNetworkInfo ( NvDsInferNetworkInfo  networkInfo)
inline

Definition at line 74 of file post_processor.h.

References m_NetworkInfo.

Field Documentation

◆ m_AllLayerInfo

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

Definition at line 116 of file post_processor.h.

◆ m_GpuID

uint32_t ModelPostProcessor::m_GpuID = 0
protected

Definition at line 112 of file post_processor.h.

◆ m_Labels

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

Definition at line 120 of file post_processor.h.

Referenced by getLabels().

◆ m_NetworkInfo

NvDsInferNetworkInfo ModelPostProcessor::m_NetworkInfo = {0}
protected

Definition at line 115 of file post_processor.h.

Referenced by setNetworkInfo().

◆ m_NetworkType

NvDsPostProcessNetworkType ModelPostProcessor::m_NetworkType = NvDsPostProcessNetworkType_Other
protected

Definition at line 109 of file post_processor.h.

◆ m_OutputLayerInfo

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

Definition at line 117 of file post_processor.h.

◆ m_UniqueID

int ModelPostProcessor::m_UniqueID = 0
protected

Definition at line 111 of file post_processor.h.

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