updates
This commit is contained in:
parent
995dc3ca67
commit
40da693ff0
|
@ -28,7 +28,7 @@ def detect(
|
||||||
|
|
||||||
# Initialize model
|
# Initialize model
|
||||||
if ONNX_EXPORT:
|
if ONNX_EXPORT:
|
||||||
s = (320, 192) # onnx model image size (height, width)
|
s = (320, 192) # (320, 192) or (416, 256) 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)
|
||||||
|
|
Loading…
Reference in New Issue