diff --git a/detect.py b/detect.py index 9106bc84..5defda55 100644 --- a/detect.py +++ b/detect.py @@ -28,7 +28,7 @@ def detect( # Initialize model if ONNX_EXPORT: - s = (320, 192) # (320, 192) or (416, 256) onnx model image size (height, width) + s = (320, 192) # (320, 192) or (416, 256) or (608, 352) onnx model image size (height, width) model = Darknet(cfg, s) else: model = Darknet(cfg, img_size)