From 95ede066c702cafe99de859a81f398568822b728 Mon Sep 17 00:00:00 2001 From: KaiyangZhou Date: Tue, 4 Sep 2018 10:11:43 +0100 Subject: [PATCH] add isinstance(src, list) check for img_paths in tracklets --- torchreid/utils/reidtools.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torchreid/utils/reidtools.py b/torchreid/utils/reidtools.py index 36d19fc..3c7dc2b 100644 --- a/torchreid/utils/reidtools.py +++ b/torchreid/utils/reidtools.py @@ -41,7 +41,7 @@ def visualize_ranked_results(distmat, dataset, save_dir='log/ranked_results', to - rank: int, denoting ranked position, starting from 1 - prefix: string """ - if isinstance(src, tuple): + if isinstance(src, tuple) or isinstance(src, list): dst = osp.join(dst, prefix + '_top' + str(rank).zfill(3)) mkdir_if_missing(dst) for img_path in src: @@ -64,4 +64,4 @@ def visualize_ranked_results(distmat, dataset, save_dir='log/ranked_results', to _cp_img_to(gimg_path, qdir, rank=rank_idx, prefix='gallery') rank_idx += 1 if rank_idx > topk: - break \ No newline at end of file + break