From 742908257a539ca829db2dcc94df995f9d4a9eb6 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 11 Feb 2019 18:17:38 +0100
Subject: [PATCH] updates

---
 detect.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/detect.py b/detect.py
index fb4c867286..67721d8db9 100755
--- 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