diff --git a/torchreid/utils/reidtools.py b/torchreid/utils/reidtools.py index 7f5b3dc..97b763f 100644 --- a/torchreid/utils/reidtools.py +++ b/torchreid/utils/reidtools.py @@ -27,7 +27,7 @@ def visualize_ranked_results(distmat, dataset, save_dir='log/ranked_results', to print("Visualizing top-{} ranks".format(topk)) print("Saving images to '{}'".format(save_dir)) - print("# query: {}\n # gallery {}".format(num_q, num_g)) + print("# query: {}\n# gallery {}".format(num_q, num_g)) assert num_q == len(dataset.query) assert num_g == len(dataset.gallery)