updates
This commit is contained in:
parent
587522ec56
commit
7e6e1897ac
|
@ -27,7 +27,7 @@ def detect(
|
||||||
|
|
||||||
# Initialize model
|
# Initialize model
|
||||||
if ONNX_EXPORT:
|
if ONNX_EXPORT:
|
||||||
s = (416, 416) # onnx model image size
|
s = (416, 416) # 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