updates
This commit is contained in:
parent
40da693ff0
commit
6c77764bba
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue