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

View File

@ -15,6 +15,8 @@ class BaseImgDataset(object):
assert isinstance(self.train, list)
assert isinstance(self.query, 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.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.query_lmdb_path, self.query)
_write_lmdb(self.gallery_lmdb_path, self.gallery)
_write_lmdb(self.gallery_lmdb_path, self.gallery)