如何将yolov5模型部署到RK3588开发板上

如何将训练好的yolov5模型(得到了训练好的pt权重)部署到RK3588开发板上:

先将训练好的模型转换为onxx格式的模型,再转换为rknn模型,最后将得到的rknn模型部署到RK3588开发板上。

1.

得到训练好的pt模型,这里为了方便,直接使用瑞芯微官方仓库的yolov5模型以及训练好的pt权重(使用瑞芯微的yolo项目能更好的适配RK3588芯片)

Yolov5模型下载地址:https://github.com/airockchip/yolov5

下载后对模型进行训练,这里为了方便直接使用官方训练好的pt权重,我们使用的是yolov5s

进入链接后在下方下载权重:

2.模型导出:

将模型导出为onnx格式的模型,首先创建自己的python虚拟环境,再进入刚才下载的yolov5项目,打开yolov5/models/yolo.py文件,将class Detection类的子函数forward函数进行修改:

将图中部分修改为:

python 复制代码
Def forward(self,x):
    Z=[]
    for i in range(self.nl):
        X[i]=self.m[i](x[i])
    return x

注意:在训练时不要修改代码,在训练完成后进行导出时修改这段代码

修完完成后将pt权重放到export.py的同一目录下,打开Anaconda prompt,激活对应python虚拟环境,进入到刚才yolo.py的目录中,运行以下代码:

python export.py --weight yolov5s.pt --img 640 --batch 1 --include onnx

然后会在目录下得到yolov5s.onnx模型文件,这代表成功将模型转换到了onnx格式

3. 安装rknn-toolkit

环境:linux系统

首先用anaconda创建python虚拟环境:

conda create -n rknn python=3.8

新建一个rknn文件夹用来存放rknn-toolkit包:

进入rknn文件夹,使用git命令下载rknn-toolkit包:

Git clone https://github.com/airockchip/rknn-toolkit2.git

下载完毕后进入rknn\rknn-toolkit2-2.1.0\rknn-toolkit2\packages

输入命令下载所需环境:

Pip install -r requirements_cp38-2.1.0.txt -i

https://pypi.tuna.tsinghua.edu.cn/simple

下载完成后进行rknn_toolkit的下载:

Pip install rknn_toolkit2-2.1.0+708089d1-cp38-cp38-linux_x86_64.whl

下载完成后进行测试rknn_toolkit是否安装成功:

在终端输入python,运行以下代码:

From rknn.api imort RKNN

如果没有报错就代表rknn_toolkit 已经安装成功

4. 将onnx模型转换为rknn格式:

进入rknn\rknn-toolkit2-2.1.0\rknn-toolkit2\exmaples\onnx\yolov5文件夹,将之前生成onnx格式的模型复制到当前目录下

修改test.py

将ONNX_MODEL修改为之前生成onnx格式的模型名,RKNN_MODEL表示生成rknn格式的模型名。

然后修改导出平台的名字:

将test.py的main函数(239行附近)中的 targrt_platform改为rk3588

然后使用命令 python test.py运行,运行成功界面如下:

这说明模型已经被转换成了rknn格式的模型

接下来只需要将rknn格式的模型部署在rk3588平台上

参考链接:

https://blog.csdn.net/m0_57315535/article/details/128250096

相关推荐
像风一样的男人@2 小时前
python --两个文件夹文件名比对(yolo 图和label标注比对检查)
windows·python·yolo
AI纪元故事会1 天前
《目标检测全解析:从R-CNN到DETR,六大经典模型深度对比与实战指南》
人工智能·yolo·目标检测·r语言·cnn
Python图像识别1 天前
75_基于深度学习的咖啡叶片病害检测系统(yolo11、yolov8、yolov5+UI界面+Python项目源码+模型+标注好的数据集)
python·深度学习·yolo
Python图像识别1 天前
74_基于深度学习的垃圾桶垃圾溢出检测系统(yolo11、yolov8、yolov5+UI界面+Python项目源码+模型+标注好的数据集)
python·深度学习·yolo
AI浩1 天前
MHAF-YOLO:用于精确目标检测的多分支异构辅助融合YOLO
人工智能·yolo·目标检测
AI视觉网奇2 天前
yolo 获取异常样本 yolo 异常
开发语言·python·yolo
FL16238631292 天前
无人机视角巡检数据集航拍建筑废物垃圾检测数据集VOC+YOLO格式3382张12类别
yolo·无人机
王哈哈^_^2 天前
【数据集】【YOLO】【目标检测】共享单车数据集,共享单车识别数据集 3596 张,YOLO自行车识别算法实战训推教程。
人工智能·算法·yolo·目标检测·计算机视觉·视觉检测·毕业设计
B站_计算机毕业设计之家2 天前
计算机视觉:python车辆行人检测与跟踪系统 YOLO模型 SORT算法 PyQt5界面 目标检测+目标跟踪 深度学习 计算机✅
人工智能·python·深度学习·算法·yolo·目标检测·机器学习
fl1768313 天前
基于yolov8+vue3实现目标检测后台管理系统
人工智能·yolo·目标检测