类说明
GetModelTensorsShape
函数功能
保护成员函数，该接口由Init()调用，用于获取模型的输入输出张量形状。
函数原型
APP_ERROR GetModelTensorsShape(MxBase::ModelDesc modelDesc);
参数说明
| 参数名 | 输入/输出 | 说明 |
|---|---|---|
| modelDesc | 输入 | 模型基本信息结构体。 |
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
CheckModelCompatibility
函数功能
保护成员函数，该接口由Init()调用，用于对所读取模型的输出形状进行校验。
函数原型
APP_ERROR CheckModelCompatibility();
返回参数说明
| 数据结构 | 说明 |
|---|---|
| APP_ERROR | 程序执行返回的错误码，请参考APP_ERROR说明。 |
类说明
该类继承了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说明。 |
在线提单