apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

updates

Browse Source
This commit is contained in:
Glenn Jocher 2019-04-29 18:02:31 +02:00
parent 587522ec56
commit 7e6e1897ac
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
detect.py
View File

@ -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)