updates

hloniphanidube · Feb 11, 2019 · 7429082 · 7429082
1 parent e23b1a3
commit 7429082
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/detect.py b/detect.py
@@ -60,7 +60,7 @@ def detect(
  img = torch.from_numpy(img).unsqueeze(0).to(device)
  if ONNX_EXPORT:
  torch.onnx.export(model, img, 'weights/model.onnx', verbose=True)
- return # ONNX export
+ return
  pred = model(img)
  pred = pred[pred[:, :, 4] > conf_thres] # remove boxes < threshold