文档
注册

总体说明

当前Python V2接口仅支持Python3.9版本。

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

基于pybind11对mxBase、mxStream进行封装,生成Python接口,实现从图像预处理、模型推理到模型后处理典型流程,以及Stream流程的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等)。
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词