updates

ultralytics · Dec 3, 2019 · 1e9ddc5 · 1e9ddc5
1 parent 0fe246f
commit 1e9ddc5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils/utils.py b/utils/utils.py
@@ -584,7 +584,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5):
                     iou = bbox_iou(dc[0], dc[1:])  # iou with other boxes
                     dc = dc[1:]
                     dc[:, 4] *= torch.exp(-iou ** 2 / sigma)  # decay confidences
-                    # dc = dc[dc[:, 4] > nms_thres]  # new line per https://github.com/ultralytics/yolov3/issues/362
+                    dc = dc[dc[:, 4] > conf_thres]  # https://github.com/ultralytics/yolov3/issues/362
 
         if len(det_max):
             det_max = torch.cat(det_max)  # concatenate