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

updates

This commit is contained in:
Glenn Jocher 2020-02-14 21:32:29 -08:00
parent 740cd177dc
commit 57798278ad
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

@ -45,7 +45,7 @@ def detect(save_img=False):
if ONNX_EXPORT: if ONNX_EXPORT:
model.fuse() model.fuse()
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=10) torch.onnx.export(model, img, 'weights/export.onnx', verbose=False, opset_version=11)
# Validate exported model # Validate exported model
import onnx import onnx