硬件设备:
本地win10工作电脑、ubuntu PC 、RK3588开发板
训练模型
在win10上首先创建Python虚拟环境并下载yolov5官方项目到本地,根据requirements.txt安装环境,在我们自己的yolov5数据集的配置文件data.yaml中追加上官方的80个类,然后正常训练数据集得到pytorch模型文件。
pt模型转onnx
在win10上创建另一个Python虚拟环境然后将RK官方的yolov5分支(以下连接)下载到本地:
根据requirements.txt安装环境,然后执行python export.py --rknpu --weight yolov5s.pt将pytorch模型转换为onnx,将onnx模型上传到ubuntu PC。
onnx转rknn
在ubuntu PC上创建Python虚拟环境,安装toolkit1.5,下载和安装步骤在以下连接中的“RKNN-Toolkit2 安装” 中:
然后下载rknn_model_zoo项目,下载连接:
airockchip/rknn_model_zoo (github.com)https://github.com/airockchip/rknn_model_zoo
在rknn_model_zoo/examples/yolov5目录下执行python convert.py ../model/yolov5s.onnx rk3588将onnx模型转为rknn模型,将rknn模型上传到rk3588开发板。
测试运行
在rk3588开发板上安装opencv3.4并下载飞桨项目,下载连接:
在FastDeploy/examples/vision/detection/rkyolo/python/目录下执行python3 infer_rkyolov5.py --model_file yolov5s.rknn --image 000000014439.jpg测试程序
飞桨环境安装部署:
opencv3.4安装文档指南:
Ubuntu20.04安装OpenCV3.4.16-CSDN博客文章浏览阅读1k次。Ubuntu20.04安装OpenCV3.4.16_ubuntu20.04安装opencv3.4.16https://blog.csdn.net/weixin_44292294/article/details/130051280Ubuntu18.04安装OpenCV依赖包libjasper-dev无法安装的问题 - 知乎近期需要在ubuntu18.04系统上安装opencv但是在安装依赖包的过程中,有一个依赖包,libjasper-dev在使用命令 sudo apt-get install libjaster-dev提示:errorE: unable to locate libjasper-dev 其中,好几个博客都…https://zhuanlan.zhihu.com/p/405446750