RK3588部署yolov5记录

硬件设备:

本地win10工作电脑、ubuntu PCRK3588开发板

训练模型

在win10上首先创建Python虚拟环境并下载yolov5官方项目到本地,根据requirements.txt安装环境,在我们自己的yolov5数据集的配置文件data.yaml中追加上官方的80个类,然后正常训练数据集得到pytorch模型文件。

pt模型转onnx

在win10上创建另一个Python虚拟环境然后将RK官方的yolov5分支(以下连接)下载到本地:

airockchip/yolov5: YOLOv5 in PyTorch > ONNX > CoreML > TFLite (github.com)icon-default.png?t=N7T8https://github.com/airockchip/yolov5

根据requirements.txt安装环境,然后执行python export.py --rknpu --weight yolov5s.pt将pytorch模型转换为onnx,将onnx模型上传到ubuntu PC。

onnx转rknn

在ubuntu PC上创建Python虚拟环境,安装toolkit1.5,下载和安装步骤在以下连接中的“RKNN-Toolkit2 安装” 中:

1. NPU使用 — Firefly Wiki (t-firefly.com)icon-default.png?t=N7T8https://wiki.t-firefly.com/zh_CN/iCore-3588Q/usage_npu.html#rknn-toolkit2-gong-ju

然后下载rknn_model_zoo项目,下载连接:

airockchip/rknn_model_zoo (github.com)icon-default.png?t=N7T8https://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并下载飞桨项目,下载连接:

PaddlePaddle/FastDeploy: ⚡️An Easy-to-use and Fast Deep Learning Model Deployment Toolkit for ☁️Cloud 📱Mobile and 📹Edge. Including Image, Video, Text and Audio 20+ main stream scenarios and 150+ SOTA models with end-to-end optimization, multi-platform and multi-framework support. (github.com)icon-default.png?t=N7T8https://github.com/PaddlePaddle/FastDeploy

在FastDeploy/examples/vision/detection/rkyolo/python/目录下执行python3 infer_rkyolov5.py --model_file yolov5s.rknn --image 000000014439.jpg测试程序

飞桨环境安装部署:

3. 技术案例 — Firefly WikiiCore-3588Q采用Rockchip RK3588新一代旗舰级八核64位处理器,最大可配32GB大内存;支持8K视频编解码;采用了BTB接口,传输能力更强;拥有丰富的接口,支持多硬盘、千兆网、WiFi6、5G/4G扩展和多种视频输入输出;支持多种操作系统;可适用于ARM PC、边缘计算、云服务器、智能NVR等领域icon-default.png?t=N7T8https://wiki.t-firefly.com/zh_CN/iCore-3588Q/linux_technical_cases.html#fei-jiang-fastdeploy)

 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 其中,好几个博客都…icon-default.png?t=N7T8https://zhuanlan.zhihu.com/p/405446750

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值