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

updates

This commit is contained in:
Glenn Jocher 2019-02-11 18:17:38 +01:00
parent e23b1a3d73
commit 742908257a
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

@ -60,7 +60,7 @@ def detect(
img = torch.from_numpy(img).unsqueeze(0).to(device) img = torch.from_numpy(img).unsqueeze(0).to(device)
if ONNX_EXPORT: if ONNX_EXPORT:
torch.onnx.export(model, img, 'weights/model.onnx', verbose=True) torch.onnx.export(model, img, 'weights/model.onnx', verbose=True)
return # ONNX export return
pred = model(img) pred = model(img)
pred = pred[pred[:, :, 4] > conf_thres] # remove boxes < threshold pred = pred[pred[:, :, 4] > conf_thres] # remove boxes < threshold