5.3 Python应用

基于Python开发应用时,可以参考我们的sophon-demo和sophon-sail两个仓库。sophon-demo仓库如上文所述,已经开源到github。sophon-sail目前只能通过SDK压缩包获取。

sophon-sail是基于Python开发应用的核心模块,它对bmlib、bmcv、ffmpeg等代码库进行了封装,并基于pybind对外提供Python接口。sophon-sail包含解码、常见图像处理、Tensor操作、推理等功能,可以搭建起完整的Python应用。

需要注意,SM9套件上默认没有安装sophon-sail的Python环境,需要从SDK目录下的sophon-sail/python_wheels/soc_BM1688/目录获取相应版本的Python whl,并在开发板上使用 pip3 install xxx.whl 命令安装。

安装完成后,可以使用 import sophon.sail as sail 命令来验证是否安装成功。

基于Python开发应用,可以参考sophon-demo仓库中各模型目录下的python例程,也可以参考sophon-sail源码目录中的sample/python目录。此目录中提供了视频编解码、yolov5/resnet等常见模型推理的参考代码。