mirrors
/
deep-person-reid
mirror of https://github.com/KaiyangZhou/deep-person-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fix Visual Rank Video Reid Error 

Fix video reid error while making sure it still works for image reid. 

When trying to save visual ranks for video reid. It crashes on statement `qdir = osp.join(save_dir, osp.basename(osp.splitext(qimg_path)[0]))` that is because qimg_path can be a tuple or list for video reid. Hence, osp.splittext gives error when input is not a str. To fix this I make sure that osp.splitext always gets pass an str regardless if we doing video or image reid.

Let me know if you have any questions.
pull/218/head
Abner Ayala-Acevedo 2019-08-23 12:12:32 -07:00 committed by GitHub
parent 0f988683fe
commit 2383698d52
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
torchreid/utils/reidtools.py
View File

@ -77,7 +77,8 @@ def visualize_ranked_results(distmat, dataset, data_type, width=128, height=256,
qimg_path, qpid, qcamid = query[q_idx]
num_cols = topk + 1
grid_img = 255 * np.ones((height, num_cols*width+topk*GRID_SPACING+QUERY_EXTRA_SPACING, 3), dtype=np.uint8)
qimg_path_name = qimg_path[0] if isinstance(qimg_path, tuple) or isinstance(qimg_path, list) else qimg_path
if data_type == 'image':
qimg = cv2.imread(qimg_path)
qimg = cv2.resize(qimg, (width, height))
@ -85,7 +86,7 @@ def visualize_ranked_results(distmat, dataset, data_type, width=128, height=256,
qimg = cv2.resize(qimg, (width, height)) # resize twice to ensure that the border width is consistent across images
grid_img[:, :width, :] = qimg
else:
qdir = osp.join(save_dir, osp.basename(osp.splitext(qimg_path)[0]))
qdir = osp.join(save_dir, osp.basename(osp.splitext(qimg_path_name)[0]))
mkdir_if_missing(qdir)
_cp_img_to(qimg_path, qdir, rank=0, prefix='query')
@ -112,7 +113,7 @@ def visualize_ranked_results(distmat, dataset, data_type, width=128, height=256,
if rank_idx > topk:
break
imname = osp.basename(osp.splitext(qimg_path)[0])
imname = osp.basename(osp.splitext(qimg_path_name)[0])
cv2.imwrite(osp.join(save_dir, imname+'.jpg'), grid_img)
if (q_idx+1) % 100 == 0: