总体说明
当前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模块:包含多种模型后处理类(ResNet-50、YOLOv3等)。
父主题: API参考(Python)V2