virtual NvDsPostProcessStatus parseEachFrame(const std::vector< NvDsInferLayerInfo > &outputLayers, NvDsPostProcessFrameOutput &result)=0
const std::vector< std::vector< std::string > > & getLabels() const
void setNetworkInfo(NvDsInferNetworkInfo networkInfo)
Holds detection and bounding box grouping parameters.
NvDsPostProcessNetworkType
Defines network types.
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 NvDsPostProcessStatus initResource(NvDsPostProcessContextInitParams &initParams)
ModelPostProcessor(NvDsPostProcessNetworkType type, int id, int gpuId)
NvDsPostProcessStatus
Enum for the status codes returned by NvDsPostProcessAlgorithm.
std::vector< std::vector< std::string > > m_Labels
NvDsInferNetworkInfo m_NetworkInfo
std::vector< NvDsInferLayerInfo > m_OutputLayerInfo
Holds the initialization parameters required for the NvDsPostProcessContext interface.
std::vector< NvDsInferLayerInfo > m_AllLayerInfo
Holds the information inferred by the network on one frame.
NvDsPostProcessStatus parseLabelsFile(const std::string &path)
Holds the output for all of the frames in a batch (an array of frame), and related buffer information...
NvDsPostProcessNetworkType m_NetworkType
virtual void releaseFrameOutput(NvDsPostProcessFrameOutput &frameOutput)=0
virtual ~ModelPostProcessor()=default
@ NvDsPostProcessNetworkType_Other
Specifies other.
void freeBatchOutput(NvDsPostProcessBatchOutput &batchOutput)