From 853bb1821def6146ae5431777bfe390b749bde0f Mon Sep 17 00:00:00 2001 From: Jilong Wang Date: Mon, 4 Sep 2023 13:17:50 +0800 Subject: [PATCH] Fix a crucial bug of mINP calculation!!! (#154) --- opengait/evaluation/metric.py | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/opengait/evaluation/metric.py b/opengait/evaluation/metric.py index 2aa43c6..42a6337 100644 --- a/opengait/evaluation/metric.py +++ b/opengait/evaluation/metric.py @@ -128,19 +128,6 @@ def evaluate_rank(distmat, p_lbls, g_lbls, max_rank=50): all_cmc.append(cmc[:max_rank]) num_valid_p += 1. - # compute average precision - # reference: https://en.wikipedia.org/wiki/Evaluation_measures_(information_retrieval)#Average_precision - num_rel = raw_cmc.sum() - pos_idx = np.where(raw_cmc == 1) # 返回坐标，此处raw_cmc为一维矩阵，所以返回相当于index - max_pos_idx = np.max(pos_idx) - inp = cmc[max_pos_idx] / (max_pos_idx + 1.0) - all_INP.append(inp) - - cmc[cmc > 1] = 1 - - all_cmc.append(cmc[:max_rank]) - num_valid_p += 1. - # compute average precision # reference: https://en.wikipedia.org/wiki/Evaluation_measures_(information_retrieval)#Average_precision num_rel = raw_cmc.sum()