update metric mAP

open-mmlab · Sep 23, 2022 · ae50a2c · ae50a2c
1 parent 2388b7f
commit ae50a2c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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