updates

2019-12-04 15:10:16 -08:00 · 2019-12-04 15:10:16 -08:00 · 54daa69adb
commit 54daa69adb
parent 5a14f54b2d
1 changed files with 1 additions and 1 deletions
--- a/detect.py
+++ b/detect.py
@ -43,7 +43,7 @@ def detect(save_txt=False, save_img=False):
    # Export mode
    if ONNX_EXPORT:
        img = torch.zeros((1, 3) + img_size)  # (1, 3, 320, 192)
-        torch.onnx.export(model, img, 'weights/export.onnx', verbose=False, opset_version=11)
+        torch.onnx.export(model, img, 'weights/export.onnx', verbose=False, opset_version=10)

        # Validate exported model
        import onnx