From 8ed0b0b97f70cf31b57aa8762ec3e88e7fc92d43 Mon Sep 17 00:00:00 2001 From: Xu GaoXiang Date: Tue, 29 Jun 2021 15:01:08 +0800 Subject: [PATCH] Update visualize_result.py --- demo/visualize_result.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/visualize_result.py b/demo/visualize_result.py index 493b702..4207a01 100644 --- 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)