From e6ce128a26600a3a4835f43a91b9dfd8edb53393 Mon Sep 17 00:00:00 2001 From: KaiyangZhou Date: Sat, 6 Oct 2018 22:16:28 +0100 Subject: [PATCH] update print --- torchreid/utils/reidtools.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchreid/utils/reidtools.py b/torchreid/utils/reidtools.py index 97b763f..a98acd6 100644 --- a/torchreid/utils/reidtools.py +++ b/torchreid/utils/reidtools.py @@ -26,8 +26,8 @@ def visualize_ranked_results(distmat, dataset, save_dir='log/ranked_results', to num_q, num_g = distmat.shape print("Visualizing top-{} ranks".format(topk)) - print("Saving images to '{}'".format(save_dir)) print("# query: {}\n# gallery {}".format(num_q, num_g)) + print("Saving images to '{}'".format(save_dir)) assert num_q == len(dataset.query) assert num_g == len(dataset.gallery)