我配置环境是
虚拟机配置了ubuntu20.4
模型是yolo26生成的.pt文件
开发板是瑞芯微的rk3576
开发板操作系统是debian12
通过下面步骤使用转换好的rknn模型在开发板上运行没有问题。一次性通过。
使用虚拟机Oracle VirtualBox安装ubuntu20.4需要注意,手工安装,让它可以选编码和语言默认为英文,会减少很多问题。
第1步.安装虚拟机的ubuntu20.4就略过,记录下怎么安装ssh服务
安装 SSH 服务端:
sudoaptupdate
sudoaptinstallopenssh-server
sudosystemctl startssh
sudosystemctlenablessh
设置让root远程登录
sudonano/etc/ssh/sshd_config
PermitRootLogin yes
sudosystemctl restartssh
将sudo passwd root密码设置为123456
第2步.在ubuntu中使用python虚拟环境转换RKNN
这个能调试通过
就一个命令就能把pt文件转换过来。
模型转换必须在 X86架构的Linux电脑(比如你的Ubuntu虚拟机)上完成,不能在开发板上进行。你需要使用rknn-toolkit2这个工具包。Ultralytics库也提供了集成的导出命令,这是最便捷的方式。
mkdir ~/rknn_workspace && cd ~/rknn_workspace
sudo apt update
sudo apt install -y python3-pip python3-venv git wget
python3 -m venv yolo_rknn_env
source yolo_rknn_env/bin/activate
# 安装 ultralytics 库
pipinstallultralytics
# 安装 rknn-toolkit2(确保你使用的是与你的开发板匹配的版本)
pipinstallrknn-toolkit2==2.3.2
pipinstallsetuptools==65.7.0
pipinstallonnx==1.13.0
在终端中直接执行以下命令。注意,这里的 format=rknn 是触发特殊导出逻辑的关键。
#yoloexportmodel=best0.96.pt format=rknn name=rk3576
转换的文件就是/root/rknn_workspace/best0.96_rknn_model/best0.96-rk3576.rknn文件。
拿到文件后,进行测试命令行代码如下:
#yolo predictmodel=./best0.96_rknn_modelsource='test1.jpg'