yolov5模型转onnx格试

yolov5模型转onnx格试

python export.py  --weights  模型文件位置  --include  onnx

例如：python export.py  --weights   runs/train/exp17/weights/best.pt   --include  onnx
转onnx格试要安装onnx
cpu安装
pip install -r requirements.txt coremltools onnx onnx-simplifier onnxruntime openvino-dev tensorflow-cpu
gpu安装
pip install -r requirements.txt coremltools onnx onnx-simplifier onnxruntime-gpu openvino-dev tensorflow

模型转onnx格试,常见错误Unsupported ONNX opset version: 17
把export.py中的onnx opset 的默认值(default = 17)从“17”改为“11”parser.add_argument('--opset', type=int, default=17, help='ONNX: opset version')大概在837行
runs/train/exp4/weights/best_test.pt