Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-02-18 19:24:53 +01:00
parent 77ce2cd43f
commit 5f2d3aa9c3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils/utils.py
View File

@ -400,12 +400,12 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.4):
elif nms_style == 'MERGE': # weighted mixture box elif nms_style == 'MERGE': # weighted mixture box
while len(dc) > 0: while len(dc) > 0:
iou = bbox_iou(dc[:1], dc[0:]) # iou with other boxes iou = bbox_iou(dc[:1], dc[0:]) # iou with other boxes
i = iou > nms_thres i = iou > .3
weights = (dc[i, 4:5] * dc[i, 5:6]) ** 2 weights = dc[i, 4:5] * dc[i, 5:6]
dc[0, :4] = (weights * dc[i, :4]).sum(0) / weights.sum() dc[0, :4] = (weights * dc[i, :4]).sum(0) / weights.sum()
det_max.append(dc[:1]) det_max.append(dc[:1])
dc = dc[iou < nms_thres] dc = dc[iou < .3]
# Image Total P R mAP # Image Total P R mAP
# 4964 5000 0.632 0.597 0.588 # normal # 4964 5000 0.632 0.597 0.588 # normal