RTMO on Axera NPU

RTMO介绍 本项目展示了RTMO模型转换和在Axera AX650 平台的部署的代码示例。
onnx模型导出参考了mmdeploy，在此基础上移除了NMS
python 推理demo参考了rtmlib，将onnx模型中移出的NMS放到了后处理过程。

• 模型转换参考 模型转换
• python demo demo

支持平台

• AX650N

模型转换

模型转换

上板部署

• AX650N 的设备已预装 Ubuntu22.04
• 以 root 权限登陆 AX650N 的板卡设备
• 链接互联网，确保 AX650N 的设备能正常执行 pip install 指令
• 已验证设备：AX650N DEMO Board、爱芯派Pro(AX650N)

Python API 运行

Requirements

安装python包

pip3 install -r python/requirments.txt

运行

记得先修改 python/infer.py中的 backend和 model_path参数
在开发板上运行命令

cd python
python3 infer.py

输入

输出

技术讨论

• Github issues
• QQ 群: 139953715