Skip to main content

Process

Init

函数功能

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

函数原型

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

参数说明

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

父主题: CtpnPostProcessor

DeInit

函数功能

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

函数原型

APP_ERROR DeInit() override;

返回参数说明

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

父主题: CtpnPostProcessor

Process

函数功能

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

函数原型

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

参数说明

参数名输入/输出说明
featLayerData输入输出特征数据数组。
textObjInfos输出输出对象坐标、置信度等信息。
useMpPictureCrop输入是否还原坐标到目标框坐标上。
postImageInfo输入图片信息(包括原图和缩放宽高,目标框坐标)。 原始和缩放宽高不能超过8192,目标框坐标不能小于0且不能超过8192。

返回参数说明

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

父主题: CtpnPostProcessor

使用须知

note

须知

本章节开放的类,请遵循同一个类实例在单一线程中使用的原则,请勿将同一个类实例在不同的线程中使用。

Python推理数据流程图

图1 推理数据流图

父主题: API参考(Python)

依赖列表说明

表1 依赖列表

依赖名称用途
Tensor与Numpy数据换转,Host/Device换转,Python模型推理的输入输出统一结构。
Image是DVPP输入输出统一结构,能转换为Tensor。
Model用于模型加载推理
ImageProcessor图像处理类,主要开放图像编解码、缩放和抠贴图等接口。
VideoDecoder视频解码类,主要开放视频解码接口。
dvpp含DVPP图像解码读取和缩放功能,即将废弃,建议使用ImageProcessor类。
log用于不同级别的日志打印。
post包含多种模型后处理类(ResNet-50、YoloV3等)。

依赖使用方法如下:

from mindx.sdk import base
from mindx.sdk.base import Image, Model, ImageProcessor, Size, post

父主题: API参考(Python)

在线提单