add isinstance(src, list) check for img_paths in tracklets
parent
7c62ab2ed8
commit
95ede066c7
|
@ -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
|
||||
break
|
||||
|
|
Loading…
Reference in New Issue