跳到主要内容

依赖列表说明

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

使用须知

备注

须知

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

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)

基本数据枚举类型

表1 数据格式枚举类型

数据类型说明
base.undefined未定义类型。
base.int8int8类型。
base.uint8uint8类型。
base.int16int16类型。
base.uint16uint16类型。
base.int32int32类型。
base.uint32uint32类型。
base.int64int64类型。
base.uint64uint64类型。
base.float16float16类型。
base.float32float32类型。
base.doubledouble类型。
base.boolbool类型。

父主题: 通用枚举类型/数据类

ModelLoadType模型加载方式枚举类型

表1 模型加载方式

参数名说明
base.load_model_from_file从文件加载离线模型数据,由系统内部管理内存。
base.load_model_from_mem从内存加载离线模型数据,由系统内部管理内存。
base.load_model_from_file_with_mem从文件加载离线模型数据,由用户自行管理模型运行的内存。
base.load_model_from_mem_with_mem从内存加载离线模型数据,由用户自行管理模型运行的内存。

父主题: 通用枚举类型/数据类

在线提单