This commit is contained in:
Glenn Jocher 2019-06-15 14:05:19 +02:00
parent 40da693ff0
commit 6c77764bba
1 changed files with 1 additions and 1 deletions

View File

@ -28,7 +28,7 @@ def detect(
# Initialize model # Initialize model
if ONNX_EXPORT: 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) model = Darknet(cfg, s)
else: else:
model = Darknet(cfg, img_size) model = Darknet(cfg, img_size)