Fix dynamic bug

naive0409 · Oct 12, 2022 · 58713d4 · 58713d4
1 parent 397d0aa
commit 58713d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deploy/ONNX/export_onnx.py b/deploy/ONNX/export_onnx.py
@@ -71,7 +71,7 @@
             'images' :{
                 0:'batch',
             },}
-        if args.end2end and not args.ort:
+        if args.end2end:
             output_axes = {
                 'num_dets': {0: 'batch'},
                 'det_boxes': {0: 'batch'},