用yolov5自带的export.py,原来的是pt模型,input image size自带是640*640
运行
python export.py --weights "C:\Users\PycharmPro
jects\py38\yolov5\models\0820.pt" --imgsz 224 --include onnx
就会生成 input image size是224的onnx模型
用yolov5自带的export.py,原来的是pt模型,input image size自带是640*640
运行
python export.py --weights "C:\Users\PycharmPro
jects\py38\yolov5\models\0820.pt" --imgsz 224 --include onnx
就会生成 input image size是224的onnx模型