mirrors/deep-person-reid
1
0
Fork 0
mirror of https://github.com/KaiyangZhou/deep-person-reid.git synced 2025-06-03 14:53:23 +08:00
Code Issues Projects Releases Wiki Activity

add isinstance(src, list) check for img_paths in tracklets

Browse Source
This commit is contained in:
KaiyangZhou 2018-09-04 10:11:43 +01:00
parent 7c62ab2ed8
commit 95ede066c7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torchreid/utils/reidtools.py
View File

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