update print

pull/119/head
KaiyangZhou 2018-10-06 22:16:28 +01:00
parent 86eb3336c9
commit e6ce128a26
1 changed files with 1 additions and 1 deletions

View File

@ -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)