mirrors/deep-person-reid
1
0
Fork 0
mirror of https://github.com/KaiyangZhou/deep-person-reid.git synced 2025-06-03 14:53:23 +08:00
Code Issues Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
KaiyangZhou 2018-03-11 21:47:45 +00:00
parent b9150adc45
commit ef99dd489a
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
eval_metrics.py
View File

@ -2,12 +2,15 @@ from __future__ import absolute_import
import numpy as np import numpy as np
import copy import copy
def evaluate(distmat, q_pids, g_pids, q_camids, g_camids): def evaluate(distmat, q_pids, g_pids, q_camids, g_camids, max_rank=50):
num_q = distmat.shape[0] num_q, num_g = distmat.shape
if num_g < max_rank:
max_rank = num_g
print("Note: number of gallery samples is quite small, got {}".format(num_g))
indices = np.argsort(distmat, axis=1) indices = np.argsort(distmat, axis=1)
matches = (g_pids[indices] == q_pids[:, np.newaxis]).astype(np.int32) matches = (g_pids[indices] == q_pids[:, np.newaxis]).astype(np.int32)
# compute cmc curve for each query, maximum rank is fixed to _MAX_RANK # compute cmc curve for each query
all_cmc = [] all_cmc = []
all_AP = [] all_AP = []
num_valid_q = 0. num_valid_q = 0.
@ -30,7 +33,7 @@ def evaluate(distmat, q_pids, g_pids, q_camids, g_camids):
cmc = cmc.cumsum() cmc = cmc.cumsum()
cmc[cmc > 1] = 1 cmc[cmc > 1] = 1
all_cmc.append(cmc) all_cmc.append(cmc[:max_rank])
num_valid_q += 1. num_valid_q += 1.
num_rel = cmc.sum() num_rel = cmc.sum()