yolov5 rknn 模型转换

yolov5 rknn 模型转换

修改yolov5的源文件

参考 https://blog.csdn.net/Chuan423425/article/details/130205729

启动转换pt 到 onnx

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

在Ubuntu启用转换 onnx 到 rknn

python3.6 安装 rknn-toolkit2 拷贝并且修改 test.py -> mytest.py

执行 mytest.py

即可完成转换