From deb9ea86e9be82ff7961d1484927f9fe5fafc16c Mon Sep 17 00:00:00 2001 From: KaiyangZhou Date: Fri, 11 May 2018 15:23:03 +0100 Subject: [PATCH] fix bug --- data_manager.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/data_manager.py b/data_manager.py index 738c485..c95babc 100755 --- a/data_manager.py +++ b/data_manager.py @@ -689,7 +689,7 @@ class iLIDSVID(object): self.dataset_url = 'http://www.eecs.qmul.ac.uk/~xiatian/iLIDS-VID/iLIDS-VID.tar' self.data_dir = osp.join(self.dataset_dir, 'i-LIDS-VID') self.split_dir = osp.join(self.dataset_dir, 'train-test people splits') - self.split_mat_path = osp.join(self.dataset_dir, 'train_test_splits_ilidsvid.mat') + self.split_mat_path = osp.join(self.split_dir, 'train_test_splits_ilidsvid.mat') self.split_path = osp.join(self.dataset_dir, 'splits.json') self.cam_1_path = osp.join(self.dataset_dir, 'i-LIDS-VID/sequences/cam1') self.cam_2_path = osp.join(self.dataset_dir, 'i-LIDS-VID/sequences/cam2') @@ -1099,4 +1099,4 @@ def init_img_dataset(name, **kwargs): def init_vid_dataset(name, **kwargs): if name not in __vid_factory.keys(): raise KeyError("Invalid dataset, got '{}', but expected to be one of {}".format(name, __vid_factory.keys())) - return __vid_factory[name](**kwargs) \ No newline at end of file + return __vid_factory[name](**kwargs)