总体说明
当前Python V2接口仅支持Python3.9版本。
基于pybind11对mxbase、mxstream进行封装,生成python接口,主要的类和模块如下:
- Tensor类:能与numpy数据换转,host/device换转,python模型推理的输入输出统一结构。
- Image类:是DVPP输入输出统一结构,能转换为tensor。
- Model类:用于模型加载推理。
- ImageProcessor类:图像处理类,主要开放图像编解码、缩放和抠贴图等接口。
- VideoDecoder类:视频解码类,主要开放视频解码接口。
- VideoEncoder类:视频编码类,主要开放视频编码接口。
- PluginNode类:插件节点,用于存储插件的基本信息,如名字、类型等。
- SequentialStream类:stream实现类,允许通过添加PluginNode的方式来构造stream。
- FunctionalStream类:stream实现类,允许以函数式构图方式来构造stream。
- dvpp模块:含Dvpp图像解码读取和缩放功能,即将废弃,建议使用ImageProcessor类。
- log模块:用于不同级别的日志打印。
- post模块:包含多种模型后处理类(Resnet50、Yolov3等)。
父主题: API参考(Python)V2