def __init__(self, root="datasets", rm_lt=False, **kwargs):
self.root = root
self.data_path = os.path.join(self.root, self.dataset_dir,
"images_convert")
self.train_label = os.path.join(self.root, self.dataset_dir,
"naic20r2_train_list_clean.txt")
self.query_label = os.path.join(self.root, self.dataset_dir,
"val_query.txt")
self.gallery_label = os.path.join(self.root, self.dataset_dir,
"val_gallery.txt")
required_files = [
self.train_label, self.query_label, self.gallery_label
]
self.check_before_run(required_files)
all_train = self.process_train(self.train_label)[:53000]
# fmt: off
if rm_lt: train = self.remove_longtail(all_train)
else: train = all_train
# fmt: on
ImageDataset.__init__(self, train, query=[], gallery=[], **kwargs)
def __init__(self, root='datasets',):
self.root = root
self.query_dir = osp.join(self.root, 'OccludedREID/query')
self.gallery_dir = osp.join(self.root, 'OccludedREID/gallery')
query, gallery = process_test(self.query_dir, self.gallery_dir)
ImageDataset.__init__(self, [], query, gallery)
def __init__(self, root='datasets',):
self.root = root
self.query_dir = osp.join(self.root, 'PartialiLIDS/query')
self.gallery_dir = osp.join(self.root, 'PartialiLIDS/gallery')
query, gallery = process_test(self.query_dir, self.gallery_dir)
ImageDataset.__init__(self, [], query, gallery)
def __init__(self, root='datasets',):
self.root = root
self.query_dir = osp.join(self.root, 'Partial_REID/partial_body_images')
self.gallery_dir = osp.join(self.root, 'Partial_REID/whole_body_images')
query, gallery = process_test(self.query_dir, self.gallery_dir)
ImageDataset.__init__(self, [], query, gallery)