未验证 提交 5a44480f 编写于 作者: F flishwnag 提交者: GitHub

Update map_utils.py (#788)

* Update map_utils.py

The calculation may be not correct when the input bbox is not ordered by prob.
This commit fixes the problem.
上级 be72f831
......@@ -105,7 +105,8 @@ class DetectionMAP(object):
# record class score positive
visited = [False] * len(gt_label)
for b in bbox:
score_idx = np.argsort(-bbox[:, 1])
for b in bbox[score_idx, :]:
label, score, xmin, ymin, xmax, ymax = b.tolist()
pred = [xmin, ymin, xmax, ymax]
max_idx = -1
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册