From f2286e7f55524c6e13748fe70c80ef99b84f80fe Mon Sep 17 00:00:00 2001
From: Jinkai Zheng <39995121+JinkaiZheng@users.noreply.github.com>
Date: Fri, 11 Jun 2021 13:46:46 +0800
Subject: [PATCH] Updated the way of reading VERI-Wild's TXT files (#507)

Reviewed by: @L1aoXingyu
---
 fastreid/data/datasets/veriwild.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fastreid/data/datasets/veriwild.py b/fastreid/data/datasets/veriwild.py
index d3cb272..ba20bcc 100644
--- 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]))