mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Updated the way of reading VERI-Wild's TXT files (#507) 

Reviewed by: @L1aoXingyu
pull/509/head
Jinkai Zheng 2021-06-11 13:46:46 +08:00 committed by GitHub
parent 4f90197336
commit f2286e7f55
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
fastreid/data/datasets/veriwild.py
View File

@ -63,7 +63,7 @@ class VeRiWild(ImageDataset):
for idx, line in enumerate(img_list_lines): for idx, line in enumerate(img_list_lines):
line = line.strip() line = line.strip()
vid = int(line.split('/')[0]) vid = int(line.split('/')[0])
imgid = line.split('/')[1] imgid = line.split('/')[1].split('.')[0]
if is_train: if is_train:
vid = self.dataset_name + '_' + str(vid) vid = self.dataset_name + '_' + str(vid)
dataset.append((self.imgid2imgpath[imgid], vid, self.imgid2camid[imgid])) dataset.append((self.imgid2imgpath[imgid], vid, self.imgid2camid[imgid]))