From 2ba45e48781ea6e34629060b7ee685a550567099 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 18 Feb 2019 19:27:34 +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 4e9fcebb..cc1707ea 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 > .3 + i = iou > .6 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 < .3] + dc = dc[iou < .6] # Image Total P R mAP # 4964 5000 0.632 0.597 0.588 # normal