From 1e9ddc5a90057d39fb47af52737dd24e99db2f07 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 2 Dec 2019 19:44:10 -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 3984d20b..ae02b9f9 100755 --- 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