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
 

你可能感兴趣的:(视觉算法部署,YOLO,人工智能,深度学习)