ubuntu22.04，瑞芯微RK3568部署YOLOv5(纯干货版)

目录

1,训练自己的数据集  转换为 onnx

2， onnx格式→rknn格式

3,3588平台部署

---

1,训练自己的数据集  转换为 onnx

在Anaconda的yolov5，进入yolov5根目录，终端运行如下

python models/export.py --weight xx.pt# 此处为需要转换模型的相对路径 当然也可以是绝对路径 如runs/train/exp/weight/best.pt
 

若出现模块不存在的情况，只需要按照requirements.txt安装即可

2， onnx格式→rknn格式

knn-toolkit2有python3.6 python3.8两个版本的包，我们这里选择python3.6（3.8我的不太行）

conda create -n rknn python=3.6
 

获取rknn-toolkit2安装包并进入改项目根目录(我这里没有用，是自己下载的包)