Merge 8ed0b0b97f into c9bc3ceb2f

2024-07-15 15:29:06 +08:00 · 2024-07-15 15:29:06 +08:00 · c848eb0004
parent c9bc3ceb2f 8ed0b0b97f
commit c848eb0004
1 changed files with 1 additions and 1 deletions
--- a/demo/visualize_result.py
+++ b/demo/visualize_result.py
@ -124,7 +124,7 @@ if __name__ == '__main__':

    # compute cosine distance
    distmat = 1 - torch.mm(q_feat, g_feat.t())
-    distmat = distmat.numpy()
+    distmat = distmat.cpu().numpy()

    logger.info("Computing APs for all query images ...")
    cmc, all_ap, all_inp = evaluate_rank(distmat, q_pids, g_pids, q_camids, g_camids)