From f788a570099a8019cb8f43d01e9488cf79a4dea8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 18 Feb 2019 19:31:00 +0100 Subject: [PATCH] updates --- utils/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/utils.py b/utils/utils.py index cc1707ea..164dfc9e 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -400,12 +400,12 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.4): elif nms_style == 'MERGE': # weighted mixture box while len(dc) > 0: iou = bbox_iou(dc[:1], dc[0:]) # iou with other boxes - i = iou > .6 + i = iou > nms_thres weights = dc[i, 4:5] * dc[i, 5:6] dc[0, :4] = (weights * dc[i, :4]).sum(0) / weights.sum() det_max.append(dc[:1]) - dc = dc[iou < .6] + dc = dc[iou < nms_thres] # Image Total P R mAP # 4964 5000 0.632 0.597 0.588 # normal