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

Detailed Description

Definition at line 30 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor_segmentation.h.

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

Public Member Functions

 SegmentationModelPostProcessor (int id, int gpuId=0)
 
 ~SegmentationModelPostProcessor () 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, NvDsRoiMeta *roi_meta, gboolean symmetric_padding) override
 
void releaseFrameOutput (NvDsPostProcessFrameOutput &frameOutput) override
 
 SegmentationModelPostProcessor (int id, int gpuId=0)
 
 ~SegmentationModelPostProcessor () 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, NvDsRoiMeta *roi_meta, gboolean symmetric_padding) override
 
void releaseFrameOutput (NvDsPostProcessFrameOutput &frameOutput) override
 
const std::vector< std::vector< std::string > > & getLabels () const
 
const std::vector< std::vector< std::string > > & getLabels () const
 
void freeBatchOutput (NvDsPostProcessBatchOutput &batchOutput)
 
void freeBatchOutput (NvDsPostProcessBatchOutput &batchOutput)
 
void setNetworkInfo (NvDsInferNetworkInfo networkInfo)
 
void setNetworkInfo (NvDsInferNetworkInfo networkInfo)
 

Protected Member Functions

NvDsPostProcessStatus parseLabelsFile (const std::string &path)
 
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

◆ SegmentationModelPostProcessor() [1/2]

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

Definition at line 33 of file sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor_segmentation.h.

◆ ~SegmentationModelPostProcessor() [1/2]

SegmentationModelPostProcessor::~SegmentationModelPostProcessor ( )
overridedefault

◆ SegmentationModelPostProcessor() [2/2]

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

Definition at line 33 of file 9.0/sources/gst-plugins/gst-nvdspostprocess/postprocesslib_impl/post_processor_segmentation.h.

◆ ~SegmentationModelPostProcessor() [2/2]

SegmentationModelPostProcessor::~SegmentationModelPostProcessor ( )
overridedefault

Member Function Documentation

◆ attachMetadata() [1/2]

void SegmentationModelPostProcessor::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 
)
overridevirtual

Implements ModelPostProcessor.

◆ attachMetadata() [2/2]

void SegmentationModelPostProcessor::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 
)
overridevirtual

Implements ModelPostProcessor.

◆ freeBatchOutput() [1/2]

void ModelPostProcessor::freeBatchOutput ( NvDsPostProcessBatchOutput &  batchOutput)
inherited

◆ freeBatchOutput() [2/2]

void ModelPostProcessor::freeBatchOutput ( NvDsPostProcessBatchOutput &  batchOutput)
inherited

◆ getLabels() [1/2]

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

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

References ModelPostProcessor::m_Labels.

◆ getLabels() [2/2]

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

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

References ModelPostProcessor::m_Labels.

◆ initResource() [1/2]

NvDsPostProcessStatus SegmentationModelPostProcessor::initResource ( NvDsPostProcessContextInitParams &  initParams)
overridevirtual

Reimplemented from ModelPostProcessor.

◆ initResource() [2/2]

NvDsPostProcessStatus SegmentationModelPostProcessor::initResource ( NvDsPostProcessContextInitParams &  initParams)
overridevirtual

Reimplemented from ModelPostProcessor.

◆ parseEachFrame() [1/2]

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

Implements ModelPostProcessor.

◆ parseEachFrame() [2/2]

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

Implements ModelPostProcessor.

◆ parseLabelsFile() [1/2]

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

◆ parseLabelsFile() [2/2]

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

◆ releaseFrameOutput() [1/2]

void SegmentationModelPostProcessor::releaseFrameOutput ( NvDsPostProcessFrameOutput &  frameOutput)
overridevirtual

Implements ModelPostProcessor.

◆ releaseFrameOutput() [2/2]

void SegmentationModelPostProcessor::releaseFrameOutput ( NvDsPostProcessFrameOutput &  frameOutput)
overridevirtual

Implements ModelPostProcessor.

◆ setNetworkInfo() [1/2]

void ModelPostProcessor::setNetworkInfo ( NvDsInferNetworkInfo  networkInfo)
inlineinherited

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

References ModelPostProcessor::m_NetworkInfo.

◆ setNetworkInfo() [2/2]

void ModelPostProcessor::setNetworkInfo ( NvDsInferNetworkInfo  networkInfo)
inlineinherited

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

References ModelPostProcessor::m_NetworkInfo.

Field Documentation

◆ m_AllLayerInfo

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

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

◆ m_GpuID

uint32_t ModelPostProcessor::m_GpuID = 0
protectedinherited

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
protectedinherited

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

Referenced by ModelPostProcessor::getLabels().

◆ m_NetworkInfo

NvDsInferNetworkInfo ModelPostProcessor::m_NetworkInfo = {0}
protectedinherited

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

Referenced by ModelPostProcessor::setNetworkInfo().

◆ m_NetworkType

NvDsPostProcessNetworkType ModelPostProcessor::m_NetworkType = NvDsPostProcessNetworkType_Other
protectedinherited

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

◆ m_OutputLayerInfo

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

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

◆ m_UniqueID

int ModelPostProcessor::m_UniqueID = 0
protectedinherited

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: