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

Reviewed by: @L1aoXingyu
2021-06-11 13:46:46 +08:00 · 2021-06-11 13:46:46 +08:00 · f2286e7f55
parent 4f90197336
commit f2286e7f55
1 changed files with 1 additions and 1 deletions
--- a/fastreid/data/datasets/veriwild.py
+++ b/fastreid/data/datasets/veriwild.py
@ -63,7 +63,7 @@ class VeRiWild(ImageDataset):
        for idx, line in enumerate(img_list_lines):
            line = line.strip()
            vid = int(line.split('/')[0])
-            imgid = line.split('/')[1]
+            imgid = line.split('/')[1].split('.')[0]
            if is_train:
                vid = self.dataset_name + '_' + str(vid)
            dataset.append((self.imgid2imgpath[imgid], vid, self.imgid2camid[imgid]))