From ae50a2ccaab27bfd6501ffbac68bc40892161bd1 Mon Sep 17 00:00:00 2001 From: zhuyu2 Date: Fri, 23 Sep 2022 16:42:03 +0800 Subject: [PATCH] update metric mAP --- mmcls/evaluation/metrics/multi_label.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mmcls/evaluation/metrics/multi_label.py b/mmcls/evaluation/metrics/multi_label.py index 0e5588896d..af4992c1b3 100644 --- a/mmcls/evaluation/metrics/multi_label.py +++ b/mmcls/evaluation/metrics/multi_label.py @@ -395,7 +395,7 @@ def _average_precision(pred: torch.Tensor, pred_pos_nums[pred_pos_nums < eps] = eps tps[torch.logical_not(pos_inds)] = 0 - precision = tps / pred_pos_nums + precision = tps / pred_pos_nums.float() ap = torch.sum(precision, 0) / max(total_pos, eps) return ap