Postprocessor object for YOLOv5Seg serials model. More...

Public Member Functions
	YOLOv5SegPostprocessor ()
	Create a postprocessor instance for YOLOv5Seg serials model.

bool	Run (const std::vector< FDTensor > &tensors, std::vector< DetectionResult > *results, const std::vector< std::map< std::string, std::array< float, 2 >>> &ims_info)
	Process the result of runtime and fill to DetectionResult structure. More...

void	SetConfThreshold (const float &conf_threshold)
	Set conf_threshold, default 0.25.

float	GetConfThreshold () const
	Get conf_threshold, default 0.25.

void	SetNMSThreshold (const float &nms_threshold)
	Set nms_threshold, default 0.5.

float	GetNMSThreshold () const
	Get nms_threshold, default 0.5.

void	SetMultiLabel (bool multi_label)
	Set multi_label, set true for eval, default true.

bool	GetMultiLabel () const
	Get multi_label, default true.

Detailed Description

Postprocessor object for YOLOv5Seg serials model.

Member Function Documentation

bool fastdeploy::vision::detection::YOLOv5SegPostprocessor::Run	(	const std::vector< FDTensor > &	tensors,
		std::vector< DetectionResult > *	results,
		const std::vector< std::map< std::string, std::array< float, 2 >>> &	ims_info
	)

Process the result of runtime and fill to DetectionResult structure.

Parameters

[in]	tensors	The inference result from runtime
[in]	result	The output result of detection
[in]	ims_info	The shape info list, record input_shape and output_shape

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

/fastdeploy/my_work/FastDeploy/fastdeploy/vision/detection/contrib/yolov5seg/postprocessor.h
/fastdeploy/my_work/FastDeploy/fastdeploy/vision/detection/contrib/yolov5seg/postprocessor.cc