yolov5转rknn,亲测成功

1 配置yolov5

下载yolov5,我下了最新版本GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

按要求装环境,requirements.txt

训练完,得到模型文件.

yolo.py第62行,加一行如下,并设置self.export=True

修改export.py文件第571行,

shape = tuple((y[0] if isinstance(y, tuple) else y).shape)

修改如下:

shape = tuple(y[0].shape)

然后将pt模型文件转为onnx文件

python export.py --weights test/best.pt --img 640 --batch 1 --include onnx --simplify --opset 12

2 best.onnx转换为best.rknn

下载rknn工程,rknn-toolkit2

按要求装环境,requirements.txt

 

 这里改一下,常规参数也填好.运行成功.