Process
类说明
该类继承了ModelPostProcessorBase用于目标检测推理后处理，用户可以根据需要继承该类并选择性实现Init、DeInit、Process等虚函数。
Init
函数功能
用于完成模型后处理初始化。
函数原型
APP_ERROR Init(const std::string& configPath, const std::string& labelPath, MxBase::ModelDesc modelDesc) override;
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| configPath | 输入 | 配置文件路径。 |
| labelPath | 输入 | 标签文件路径。 |
| modelDesc | 输出 | 模型描述信息。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
DeInit
函数功能
用于模型后处理的去初始化，完成资源释放。
函数原型
APP_ERROR DeInit() override;
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
Process
函数功能
从输出数据中获取检测目标的信息。
函数原型
virtual APP_ERROR Process(std::vector<std::shared_ptr<void>>& featLayerData, std::vector<ObjDetectInfo>& objInfos,const bool useMpPictureCrop, MxBase::PostImageInfo postImageInfo);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| featLayerData | 输入 | 输出特征数据数组。 |
| objInfos | 输出 | 输出对象坐标、类型、置信度等信息。 输出单个对象的坐标范围为[0, 8192]，类型不能小于0。 |
| useMpPictureCrop | 输入 | 是否还原坐标到目标框坐标上。 |
| postImageInfo | 输入 | 图片信息（包括原图和缩放宽高，目标框坐标）。原图和缩放宽高均不能超过8192，目标框坐标范围为[0, 8192]。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
SetAspectRatioImageInfo
函数功能
设置后处理图像信息。
函数原型
void SetAspectRatioImageInfo(const MxBase::AspectRatioPostImageInfo& postProcessorImageInfo);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| postProcessorImageInfo | 输入 | 后处理图像信息。 |
ObjectDetectionOutput
函数功能
保护成员函数，由Process()调用，用于对模型输出进行目标处理，如去重、排序、筛选等。该方法为内部调用方法，请用户不要使用。
函数原型
virtual void ObjectDetectionOutput(std::vector<std::shared_ptr<void>>& featLayerData,
std::vector<ObjDetectInfo>& objInfos, ImageInfo& imgInfo);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| featLayerData | 输入 | 输出特征数据数组。 |
| objInfos | 输出 | 输出对象坐标、类型、置信度等信息。 |
| imgInfo | 输入 | 图片信息。 |
CoordinatesReduction
函数功能
保护成员函数，用于对坐标数据进行缩放。该方法为内部调用方法，请用户不要使用。
函数原型
void CoordinatesReduction(ImageInfo& imgInfo, std::vector<ObjDetectInfo>& objInfos);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| objInfos | 输出 | 输出目标检测的坐标、类型、置信度等信息。 |
| imageInfo | 输入 | 图片信息（包括原图和缩放后的宽高）。 |
在线提单