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
即可完成转换