Fix ONNX opset inconsistency with parseargs and run args (ultralytics…

…#5937)
fhkiel-mlaip · Dec 9, 2021 · ad2571f · ad2571f
1 parent 5b8d52a
commit ad2571f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/export.py b/export.py
@@ -325,7 +325,7 @@ def run(data=ROOT / 'data/coco128.yaml',  # 'dataset.yaml path'
         int8=False,  # CoreML/TF INT8 quantization
         dynamic=False,  # ONNX/TF: dynamic axes
         simplify=False,  # ONNX: simplify model
-        opset=12,  # ONNX: opset version
+        opset=14,  # ONNX: opset version
         verbose=False,  # TensorRT: verbose log
         workspace=4,  # TensorRT: workspace size (GB)
         topk_per_class=100,  # TF.js NMS: topk per class to keep