diff --git a/detect.py b/detect.py index 889ca689..d3eff739 100644 --- a/detect.py +++ b/detect.py @@ -45,7 +45,7 @@ def detect(save_img=False): if ONNX_EXPORT: model.fuse() img = torch.zeros((1, 3) + img_size) # (1, 3, 320, 192) - torch.onnx.export(model, img, 'weights/export.onnx', verbose=False, opset_version=10) + torch.onnx.export(model, img, 'weights/export.onnx', verbose=False, opset_version=11) # Validate exported model import onnx