mirrors/fast-reid
1
0
Fork 0
mirror of https://github.com/JDAI-CV/fast-reid.git synced 2025-06-03 14:50:47 +08:00
Code Issues Projects Releases Wiki Activity

Merge 8ed0b0b97f70cf31b57aa8762ec3e88e7fc92d43 into c9bc3ceb2f7a6438b62fb515ea3df6d1e999e95d

Browse Source
This commit is contained in:
Xu GaoXiang 2024-07-15 15:29:06 +08:00 committed by GitHub
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)