diff --git a/utils/export_codetr.py b/utils/export_codetr.py index c963b14d..25f292a8 100644 --- a/utils/export_codetr.py +++ b/utils/export_codetr.py @@ -111,7 +111,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_damoyolo.py b/utils/export_damoyolo.py index 40269f56..0a9d2f86 100644 --- a/utils/export_damoyolo.py +++ b/utils/export_damoyolo.py @@ -80,7 +80,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_dfine.py b/utils/export_dfine.py index 853c3ae5..e353346f 100644 --- a/utils/export_dfine.py +++ b/utils/export_dfine.py @@ -76,7 +76,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_goldyolo.py b/utils/export_goldyolo.py index 2c41d849..caac3d4b 100644 --- a/utils/export_goldyolo.py +++ b/utils/export_goldyolo.py @@ -86,7 +86,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_rtdetr_pytorch.py b/utils/export_rtdetr_pytorch.py index 9c3dc4b2..4fd159e3 100755 --- a/utils/export_rtdetr_pytorch.py +++ b/utils/export_rtdetr_pytorch.py @@ -74,7 +74,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_rtdetr_ultralytics.py b/utils/export_rtdetr_ultralytics.py index 3f5eca86..57da68ec 100755 --- a/utils/export_rtdetr_ultralytics.py +++ b/utils/export_rtdetr_ultralytics.py @@ -84,7 +84,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_rtmdet.py b/utils/export_rtmdet.py index 1b7420d4..cf191635 100644 --- a/utils/export_rtmdet.py +++ b/utils/export_rtmdet.py @@ -112,7 +112,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yolo11.py b/utils/export_yolo11.py index ebb693c0..42d11022 100644 --- a/utils/export_yolo11.py +++ b/utils/export_yolo11.py @@ -106,7 +106,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yoloV10.py b/utils/export_yoloV10.py index 6bda9d41..ed9f1ce0 100644 --- a/utils/export_yoloV10.py +++ b/utils/export_yoloV10.py @@ -121,7 +121,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yoloV5.py b/utils/export_yoloV5.py index e7dbafe7..39013a31 100644 --- a/utils/export_yoloV5.py +++ b/utils/export_yoloV5.py @@ -81,7 +81,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yoloV5u.py b/utils/export_yoloV5u.py index 9cc50c9e..275cb169 100644 --- a/utils/export_yoloV5u.py +++ b/utils/export_yoloV5u.py @@ -106,7 +106,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yoloV6.py b/utils/export_yoloV6.py index 26887f5d..384d3689 100644 --- a/utils/export_yoloV6.py +++ b/utils/export_yoloV6.py @@ -87,7 +87,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yoloV7.py b/utils/export_yoloV7.py index 23e67608..21e8e690 100644 --- a/utils/export_yoloV7.py +++ b/utils/export_yoloV7.py @@ -87,7 +87,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yoloV7_u6.py b/utils/export_yoloV7_u6.py index 35eae247..5571cff2 100644 --- a/utils/export_yoloV7_u6.py +++ b/utils/export_yoloV7_u6.py @@ -78,7 +78,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yoloV8.py b/utils/export_yoloV8.py index fc5e8fbc..1578d819 100644 --- a/utils/export_yoloV8.py +++ b/utils/export_yoloV8.py @@ -106,7 +106,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yoloV9.py b/utils/export_yoloV9.py index 0d088bbe..9b1fded1 100644 --- a/utils/export_yoloV9.py +++ b/utils/export_yoloV9.py @@ -110,7 +110,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yolonas.py b/utils/export_yolonas.py index d36c97ae..7e5da8ce 100644 --- a/utils/export_yolonas.py +++ b/utils/export_yolonas.py @@ -62,7 +62,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yolor.py b/utils/export_yolor.py index 63805040..1de3ac84 100644 --- a/utils/export_yolor.py +++ b/utils/export_yolor.py @@ -98,7 +98,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: diff --git a/utils/export_yolox.py b/utils/export_yolox.py index 902bd5e1..4130267c 100644 --- a/utils/export_yolox.py +++ b/utils/export_yolox.py @@ -75,7 +75,8 @@ def main(args): print('Exporting the model to ONNX') torch.onnx.export( model, onnx_input_im, onnx_output_file, verbose=False, opset_version=args.opset, do_constant_folding=True, - input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None + input_names=['input'], output_names=['output'], dynamic_axes=dynamic_axes if args.dynamic else None, + dynamo=False ) if args.simplify: