Skip to main content

ModelLoadType模型加载方式枚举类型

依赖列表说明

表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从内存加载离线模型数据,由用户自行管理模型运行的内存。

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

ModelType模型枚举类型

表1 模型类型

参数名说明
base.model_type_omOM模型。
base.model_type_mindirMindIR模型(对于MindIR模型,仅支持静态Shape和动态Batch。)。

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

VisionDataFormat图像数据格式排布枚举类型

表1 图像数据格式排布形式

排布形式说明
NCHW图像数据按NCHW格式排布。
NHWC图像数据按NHWC格式排布。

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

在线提单