From bf66656b4e8ae59a29c478d69394eb6571dbb2b8 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Fri, 23 Nov 2018 15:34:49 +0100
Subject: [PATCH] updates

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

diff --git a/utils/utils.py b/utils/utils.py
index 990cc115..9ae9b931 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -306,7 +306,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.4):
             # thresh = 0.85
             thresh = nms_thres
             a = pred.clone()
-            _, indices = torch.sort(-a[:, 4], 0) # sort best to worst
+            _, indices = torch.sort(-a[:, 4], 0)  # sort best to worst
             a = a[indices]
             radius = 30  # area to search for cross-class ious
             for i in range(len(a)):