fix: [RT-DETRv2][export_onnx] fixed args typo: file_name -> output_fi…

…le (#542)
lyuwenyu · Feb 2, 2025 · 1ced664 · 1ced664
1 parent 5ca3645
commit 1ced664
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rtdetrv2_pytorch/tools/export_onnx.py b/rtdetrv2_pytorch/tools/export_onnx.py
@@ -76,8 +76,8 @@ def forward(self, images, orig_target_sizes):
         dynamic = True 
         # input_shapes = {'images': [1, 3, 640, 640], 'orig_target_sizes': [1, 2]} if dynamic else None
         input_shapes = {'images': data.shape, 'orig_target_sizes': size.shape} if dynamic else None
-        onnx_model_simplify, check = onnxsim.simplify(args.file_name, input_shapes=input_shapes, dynamic_input_shape=dynamic)
-        onnx.save(onnx_model_simplify, args.file_name)
+        onnx_model_simplify, check = onnxsim.simplify(args.output_file, input_shapes=input_shapes, dynamic_input_shape=dynamic)
+        onnx.save(onnx_model_simplify, args.output_file)
         print(f'Simplify onnx model {check}...')