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()