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

add reminder

pull/62/head
KaiyangZhou 2018-07-02 15:11:27 +01:00
parent 94941fe69a
commit 2f320090a6
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
data_manager/base.py
View File

@ -15,6 +15,8 @@ class BaseImgDataset(object):
assert isinstance(self.train, list) assert isinstance(self.train, list)
assert isinstance(self.query, list) assert isinstance(self.query, list)
assert isinstance(self.gallery, list) assert isinstance(self.gallery, list)
print("Reminder: this function is under development, some datasets might not be applicable yet")
self.train_lmdb_path = osp.join(self.dataset_dir, 'train_lmdb') self.train_lmdb_path = osp.join(self.dataset_dir, 'train_lmdb')
self.query_lmdb_path = osp.join(self.dataset_dir, 'query_lmdb') self.query_lmdb_path = osp.join(self.dataset_dir, 'query_lmdb')
@ -38,4 +40,4 @@ class BaseImgDataset(object):
_write_lmdb(self.train_lmdb_path, self.train) _write_lmdb(self.train_lmdb_path, self.train)
_write_lmdb(self.query_lmdb_path, self.query) _write_lmdb(self.query_lmdb_path, self.query)
_write_lmdb(self.gallery_lmdb_path, self.gallery) _write_lmdb(self.gallery_lmdb_path, self.gallery)