Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-12-04 15:10:16 -08:00
parent 5a14f54b2d
commit 54daa69adb
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

@ -43,7 +43,7 @@ def detect(save_txt=False, save_img=False):
# Export mode # Export mode
if ONNX_EXPORT: if ONNX_EXPORT:
img = torch.zeros((1, 3) + img_size) # (1, 3, 320, 192) img = torch.zeros((1, 3) + img_size) # (1, 3, 320, 192)
torch.onnx.export(model, img, 'weights/export.onnx', verbose=False, opset_version=11) torch.onnx.export(model, img, 'weights/export.onnx', verbose=False, opset_version=10)
# Validate exported model # Validate exported model
import onnx import onnx