若该文为原创文章,转载请注明原文出处。
一、介绍
实现的目标是,把RK提供的yolov5s.pt转成onnx,在把onnx转成rknn,部署到RV1126上面。
这里不训练模型,所以只要搭建好环境后,就可以直接运行测试。
这里只是提供一种转换的方法,有其他的方式,可以自行测试。
由于不训练模型,所以所有的操作直接在虚拟机下进行,操作分为三部分:一环境搭建,二、文件下载;三是模型如何转换的。
三、环境搭建
注意,注意,注意,重要的事强调三变。
注意:pt转onnx用py3.8,onnx转rknn用py3.6,一定要有开发板,转成rknn需要连接仿真测试,衣可以模拟,但精度会降低(亲测使用的是正点原子的开发板)。
1、miniconda安装
下载地址
https://docs.conda.io/en/latest/miniconda.html
下载后直接运行下面命令安装,要先授权
chmod +x Miniconda3-py38_4.12.0-Linux-x86_64.sh
bash Miniconda3-py38_