crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix a crucial bug of mINP calculation!!! (#154)

Browse Source
This commit is contained in:
Jilong Wang
2023-09-04 13:17:50 +08:00
committed by GitHub
parent f280be5a3f
commit 853bb1821d
1 changed files with 0 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
opengait/evaluation/metric.py
-13
View File
@@ -128,19 +128,6 @@ def evaluate_rank(distmat, p_lbls, g_lbls, max_rank=50):
all_cmc.append(cmc[:max_rank]) all_cmc.append(cmc[:max_rank])
num_valid_p += 1. 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 # compute average precision
# reference: https://en.wikipedia.org/wiki/Evaluation_measures_(information_retrieval)#Average_precision # reference: https://en.wikipedia.org/wiki/Evaluation_measures_(information_retrieval)#Average_precision
num_rel = raw_cmc.sum() num_rel = raw_cmc.sum()