怎么将yolo26的.pt文件转换为3576的.rknn文件
2026/9/8 16:37:04 网站建设 项目流程

我配置环境是

虚拟机配置了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'

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询