From 723431d6c3272409559e2e7138395807583362af Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sun, 19 Jan 2020 12:15:42 -0800
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 b53fa9d7..21b42e76 100755
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -513,7 +513,7 @@ def non_max_suppression(prediction, conf_thres=0.5, iou_thres=0.5, multi_cls=Tru
 
     method = 'vision_batch'
     nc = prediction[0].shape[1] - 5  # number of classes
-    multi_cls = multi_cls and (nc > 1)
+    multi_cls = multi_cls and (nc > 1)  # allow multiple classes per anchor
     output = [None] * len(prediction)
     for image_i, pred in enumerate(prediction):
         # Apply conf constraint