mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge 8ed0b0b97f into c9bc3ceb2f

pull/526/merge
Xu GaoXiang 2024-07-15 15:29:06 +08:00 committed by GitHub
parent c9bc3ceb2f 8ed0b0b97f
commit c848eb0004
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
demo/visualize_result.py
View File

@ -124,7 +124,7 @@ if __name__ == '__main__':
# compute cosine distance # compute cosine distance
distmat = 1 - torch.mm(q_feat, g_feat.t()) distmat = 1 - torch.mm(q_feat, g_feat.t())
distmat = distmat.numpy() distmat = distmat.cpu().numpy()
logger.info("Computing APs for all query images ...") logger.info("Computing APs for all query images ...")
cmc, all_ap, all_inp = evaluate_rank(distmat, q_pids, g_pids, q_camids, g_camids) cmc, all_ap, all_inp = evaluate_rank(distmat, q_pids, g_pids, q_camids, g_camids)