在利用yolov5中的models/export.py文件转换训练好的模型best.pt时,遇到了报错:export.py: error: unrecognized arguments: --include onnx
将best.pt和export.py放在yolov5同一目录下,输入以下指令:
python export.py --weights best.pt --img 640 --batch 1 --include onnx
发现报错export.py: error: unrecognized arguments: --include onnx,大意是指有无法识别的参数,并且给出了提示:–include onnx
因而删除掉–include onnx,执行以下指令,即可成功生成best.onnx文件。
python export.py --weights best.pt --img 640 --batch 1
我用来训练模型的yolov5不是官方文档,而是更改后适配于rk3588平台的文档,链接在此
里面有更改后的yolov5/v7/v8等文档,和相应开发平台部署教程可供参考。