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

Detailed Description

Definition at line 57 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/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, NvDsRoiMeta *roi_meta, gboolean symmetric_padding)=0
 
virtual void releaseFrameOutput (NvDsPostProcessFrameOutput &frameOutput)=0
 
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, NvDsRoiMeta *roi_meta, gboolean symmetric_padding)=0
 
virtual void releaseFrameOutput (NvDsPostProcessFrameOutput &frameOutput)=0
 

Protected Member Functions

 ModelPostProcessor (NvDsPostProcessNetworkType type, int id, int gpuId)
 
NvDsPostProcessStatus parseLabelsFile (const std::string &path)
 
 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() [1/2]

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

Definition at line 61 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

◆ ~ModelPostProcessor() [1/2]

virtual ModelPostProcessor::~ModelPostProcessor ( )
virtualdefault

◆ ModelPostProcessor() [2/2]

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

Definition at line 61 of file 9.0/sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

◆ ~ModelPostProcessor() [2/2]

virtual ModelPostProcessor::~ModelPostProcessor ( )
virtualdefault

Member Function Documentation

◆ attachMetadata() [1/2]

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,
NvDsRoiMeta *  roi_meta,
gboolean  symmetric_padding 
)
pure virtual

Implemented in BodyPoseModelPostProcessor, BodyPoseModelPostProcessor, InstanceSegmentModelPostProcessor, InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, DetectModelPostProcessor, SegmentationModelPostProcessor, and SegmentationModelPostProcessor.

◆ attachMetadata() [2/2]

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,
NvDsRoiMeta *  roi_meta,
gboolean  symmetric_padding 
)
pure virtual

Implemented in BodyPoseModelPostProcessor, BodyPoseModelPostProcessor, InstanceSegmentModelPostProcessor, InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, DetectModelPostProcessor, SegmentationModelPostProcessor, and SegmentationModelPostProcessor.

◆ freeBatchOutput() [1/2]

void ModelPostProcessor::freeBatchOutput ( NvDsPostProcessBatchOutput &  batchOutput)

◆ freeBatchOutput() [2/2]

void ModelPostProcessor::freeBatchOutput ( NvDsPostProcessBatchOutput &  batchOutput)

◆ getLabels() [1/2]

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

Definition at line 69 of file 9.0/sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

References m_Labels.

◆ getLabels() [2/2]

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

Definition at line 69 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

References m_Labels.

◆ initResource() [1/2]

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

Reimplemented in BodyPoseModelPostProcessor, BodyPoseModelPostProcessor, InstanceSegmentModelPostProcessor, InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, DetectModelPostProcessor, SegmentationModelPostProcessor, and SegmentationModelPostProcessor.

◆ initResource() [2/2]

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

Reimplemented in BodyPoseModelPostProcessor, BodyPoseModelPostProcessor, InstanceSegmentModelPostProcessor, InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, DetectModelPostProcessor, SegmentationModelPostProcessor, and SegmentationModelPostProcessor.

◆ parseEachFrame() [1/2]

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

Implemented in BodyPoseModelPostProcessor, BodyPoseModelPostProcessor, InstanceSegmentModelPostProcessor, InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, DetectModelPostProcessor, SegmentationModelPostProcessor, and SegmentationModelPostProcessor.

◆ parseEachFrame() [2/2]

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

Implemented in BodyPoseModelPostProcessor, BodyPoseModelPostProcessor, InstanceSegmentModelPostProcessor, InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, DetectModelPostProcessor, SegmentationModelPostProcessor, and SegmentationModelPostProcessor.

◆ parseLabelsFile() [1/2]

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

◆ parseLabelsFile() [2/2]

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

◆ releaseFrameOutput() [1/2]

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

Implemented in BodyPoseModelPostProcessor, BodyPoseModelPostProcessor, InstanceSegmentModelPostProcessor, InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, DetectModelPostProcessor, SegmentationModelPostProcessor, and SegmentationModelPostProcessor.

◆ releaseFrameOutput() [2/2]

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

Implemented in BodyPoseModelPostProcessor, BodyPoseModelPostProcessor, InstanceSegmentModelPostProcessor, InstanceSegmentModelPostProcessor, ClassifyModelPostProcessor, ClassifyModelPostProcessor, DetectModelPostProcessor, DetectModelPostProcessor, SegmentationModelPostProcessor, and SegmentationModelPostProcessor.

◆ setNetworkInfo() [1/2]

void ModelPostProcessor::setNetworkInfo ( NvDsInferNetworkInfo  networkInfo)
inline

Definition at line 74 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

References m_NetworkInfo.

◆ setNetworkInfo() [2/2]

void ModelPostProcessor::setNetworkInfo ( NvDsInferNetworkInfo  networkInfo)
inline

Definition at line 74 of file 9.0/sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

References m_NetworkInfo.

Field Documentation

◆ m_AllLayerInfo

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

Definition at line 118 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

◆ m_GpuID

uint32_t ModelPostProcessor::m_GpuID = 0
protected

Definition at line 114 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

◆ m_Labels

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

Definition at line 122 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

Referenced by getLabels().

◆ m_NetworkInfo

NvDsInferNetworkInfo ModelPostProcessor::m_NetworkInfo = {0}
protected

Definition at line 117 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

Referenced by setNetworkInfo().

◆ m_NetworkType

NvDsPostProcessNetworkType ModelPostProcessor::m_NetworkType = NvDsPostProcessNetworkType_Other
protected

Definition at line 111 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

◆ m_OutputLayerInfo

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

Definition at line 119 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

◆ m_UniqueID

int ModelPostProcessor::m_UniqueID = 0
protected

Definition at line 113 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor.h.

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