跳到主要内容

Init

NmsSort

函数功能

保护成员函数,根据置信度大小过滤重复目标。该方法为内部调用方法,请用户不要使用。

函数原型

void NmsSort(std::vector<ObjDetectInfo>& objInfos, float iouThresh, IOUMethod method = UNION);

参数说明

参数名输入/输出说明
objInfos输出待过滤的目标边角点。
iouThresh输入过滤阀值。
method输入IOU计算方式:
- MAX 重叠区域除以两者中面积大的。
- MIN 重叠区域除以两者中面积小的。
- UNION 重叠区域除以两者面积并集。
- DIOU 重叠区域除以两者面积并集减去距离的交并比。

父主题: ObjectPostProcessorBase

类说明

该类继承了ModelPostProcessorBase用于FasterRcnn模型后处理,用户可以根据需要继承该类并选择性实现Init、DeInit、Process等虚函数。

父主题: FasterRcnnPostProcessor

Init

函数功能

用于完成模型后处理初始化。

函数原型

APP_ERROR Init(const std::string &configPath, const std::string &labelPath, MxBase::ModelDesc modelDesc) override;

参数说明

参数名输入/输出说明
configPath输入配置文件路径。
labelPath输入标签文件路径。
modelDesc输出模型描述信息。

返回参数说明

数据结构说明
APP_ERROR程序执行返回的错误码,请参考APP_ERROR说明

父主题: FasterRcnnPostProcessor

DeInit

函数功能

用于模型后处理的去初始化,完成资源释放。

函数原型

APP_ERROR DeInit() override;

返回参数说明

数据结构说明
APP_ERROR程序执行返回的错误码,请参考APP_ERROR说明

父主题: FasterRcnnPostProcessor

Process

函数功能

从输出数据中获取检测目标的信息。

函数原型

APP_ERROR Process(std::vector<std::shared_ptr<void>> &featLayerData, std::vector<ObjDetectInfo> &objInfos,const bool useMpPictureCrop, MxBase::PostImageInfo postImageInfo);

参数说明

参数名输入/输出说明
featLayerData输入输出特征数据数组。
objInfos输出输出对象坐标、类型、置信度等信息。
useMpPictureCrop输入是否还原坐标到目标框坐标上。
postImageInfo输入图片信息(包括原图和缩放宽高,目标框坐标)。

返回参数说明

数据结构说明
APP_ERROR程序执行返回的错误码,请参考APP_ERROR说明

父主题: FasterRcnnPostProcessor

类说明

该类继承了ModelPostProcessorBase用于YOLOv3模型后处理,用户可以根据需要继承该类并选择性实现Init、DeInit、Process等虚函数。

父主题: Yolov3PostProcessor

在线提单