模型修改input 的img size
用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模型。