def get_helen_test_data(query_label_names, aug_setting_name):
return ps.Dataset(
'HELENRelabeled_wo_pred',
category='test',
aug_ids=[0],
aug_setting_name=aug_setting_name,
query_label_names=query_label_names)
def gen_training_data(self,
query_label_names,
aug_setting_name='aug_512_0.8',
dataset_names=[]):
datasets = []
if len(dataset_names) == 0:
dataset_names = [
'HELENRelabeled', 'MultiPIE', 'HangYang', 'Portrait724'
]
for dataset_name in dataset_names:
datasets.append(
ps.Dataset(dataset_name,
category='train',
aug_ids=[0, 1, 2, 3],
aug_setting_name=aug_setting_name,
query_label_names=query_label_names))
return ps.CombinedDataset(datasets)
