RV1126笔记二十二:pt->onnx->rknn模型转换

若该文为原创文章,转载请注明原文出处。

一、介绍

实现的目标是,把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_

猜你喜欢

转载自blog.csdn.net/weixin_38807927/article/details/128439992