mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix bug: add return

pull/119/head
KaiyangZhou 2018-11-10 21:16:31 +00:00
parent d885f11ce7
commit 41f1f6c6ac
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
torchreid/eval_metrics.py
View File

@ -17,10 +17,10 @@ except ImportError:
warnings.warn("Cython evaluation is UNAVAILABLE, which is highly recommended")
def eval_cuhk03(distmat, q_pids, g_pids, q_camids, g_camids, max_rank, N=100):
def eval_cuhk03(distmat, q_pids, g_pids, q_camids, g_camids, max_rank, N=10):
"""Evaluation with cuhk03 metric
Key: one image for each gallery identity is randomly sampled for each query identity.
Random sampling is performed N times (default: N=100).
Random sampling is performed N times.
"""
num_q, num_g = distmat.shape
if num_g < max_rank:
@ -148,4 +148,4 @@ def evaluate(distmat, q_pids, g_pids, q_camids, g_camids, max_rank=50, use_metri
if use_metric_cuhk03:
return eval_cuhk03(distmat, q_pids, g_pids, q_camids, g_camids, max_rank)
else:
eval_market1501(distmat, q_pids, g_pids, q_camids, g_camids, max_rank)
return eval_market1501(distmat, q_pids, g_pids, q_camids, g_camids, max_rank)