updates
This commit is contained in:
@@ -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)
|
||||||
|
|||||||
Reference in New Issue
Block a user