From 54daa69adb48f4766f33c274740dcfe43957f808 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 4 Dec 2019 15:10:16 -0800 Subject: [PATCH] updates --- detect.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/detect.py b/detect.py index 5a7fb59b..2593406c 100644 --- 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