clipart2VOC(split, year, devkit_path=devkit_path))
# Set up pl_clipart voc_<year>_<split>
for year in ['2007', '2012']:
devkit_path = '/userhome/Datasets/pl_clipart'
for split in ['train', 'val', 'trainval', 'test']:
name = 'pl_clipart_voc_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year, devkit_path=devkit_path:
pl_clipart(split, year, devkit_path=devkit_path))
# Set up comic_<year>_<split>
for year in ['2007']:
devkit_path = '/userhome/Datasets/comic'
for split in ['train', 'test', 'trainval', 'train_cg']:
name = 'comic_{}_{}'.format(year, split)
__sets[name] = (lambda year=year, split=split, devkit_path=devkit_path:
comic(split, year, devkit_path=devkit_path))
# Set up dt_comic voc_<year>_<split>
for year in ['2007', '2012']:
devkit_path = '/userhome/Datasets/dt_comic'
for split in ['train', 'val', 'trainval', 'test']:
name = 'dt_comic_voc_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year, devkit_path=devkit_path:
dt_comic(split, year, devkit_path=devkit_path))
# Set up comic2VOC_<year>_<split>
for year in ['2007']:
devkit_path = '/userhome/Datasets/comic2VOC'
for split in ['train', 'test', 'trainval']:
name = 'comic2VOC_{}_{}'.format(year, split)
__sets[name] = (lambda year=year, split=split, devkit_path=devkit_path:
comic2VOC(split, year, devkit_path=devkit_path))
# Set up pl_comic_<year>_<split>
# AMD test splits
for split in ['test']:
data_percentage_split = ''
name = 'clipart{}_{}'.format(data_percentage_split, split)
__sets[name] = (
lambda split=split, data_percentage_split=data_percentage_split:
clipart('clipart{}'.format(data_percentage_split),
split,
devkit_path=os.path.join(
'datasets/', 'clipart{}'.format(data_percentage_split))))
name = 'comic{}_{}'.format(data_percentage_split, split)
__sets[name] = (
lambda split=split, data_percentage_split=data_percentage_split: comic(
'comic{}'.format(data_percentage_split),
split,
devkit_path=os.path.join('datasets/', 'comic{}'.format(
data_percentage_split))))
name = 'watercolor{}_{}'.format(data_percentage_split, split)
__sets[name] = (
lambda split=split, data_percentage_split=data_percentage_split:
watercolor('watercolor{}'.format(data_percentage_split),
split,
devkit_path=os.path.join(
'datasets/', 'watercolor{}'.format(data_percentage_split
))))
# AMD train splits
for split in ['train']:
# Set up clipart
for split in [
'all', 'all_with_label', 'all_without_label', 'all_with_label_AL'
]:
name = 'clipart_{}'.format(split)
__sets[name] = (lambda split=split, year=year: clipart(split))
# Set up watercolor
for split in ['train_with_label', 'train_without_label', 'test']:
name = 'watercolor_{}'.format(split)
__sets[name] = (lambda split=split, year=year: watercolor(split))
# Set up comic
for split in ['train_with_label', 'train_without_label', 'test']:
name = 'comic_{}'.format(split)
__sets[name] = (lambda split=split, year=year: comic(split))
# Set up coco_2014_<split>
for year in ['2014']:
for split in ['train', 'val', 'minival', 'valminusminival', 'trainval']:
name = 'coco_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: coco(split, year))
# # Set up coco_2014_cap_<split>
# for year in ['2014']:
# for split in ['train', 'val', 'capval', 'valminuscapval', 'trainval']:
# name = 'coco_{}_{}'.format(year, split)
# __sets[name] = (lambda split=split, year=year: coco(split, year))
#
# # Set up coco_2015_<split>
# for year in ['2015']:
name = 'voc_cycleclipart_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: pascal_voc_cycleclipart(split, year))
for year in ['2007', '2012']:
for split in ['train', 'val', 'trainval', 'test']:
name = 'voc_cyclewater_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: pascal_voc_cyclewater(split, year))
for split in ['train', 'trainval', 'test']:
for data_percentage in ['', '_1_00', '_1_01', '_1_02']:
dataset_name = 'clipart{}'.format(data_percentage)
name = '{}_{}'.format(dataset_name, split)
__sets[name] = (lambda split=split, dataset_name=dataset_name: clipart(split,devkit_path=os.path.join('datasets', dataset_name)))
for split in ['train', 'test']:
for data_percentage in ['', '_1_00', '_1_01', '_1_02']:
dataset_name = 'comic{}'.format(data_percentage)
name = '{}_{}'.format(dataset_name, split)
__sets[name] = (lambda split=split, dataset_name=dataset_name: comic(split,devkit_path=os.path.join('datasets', dataset_name)))
for split in ['train', 'test']:
for data_percentage in ['', '_1_00', '_1_01', '_1_02']:
dataset_name = 'watercolor{}'.format(data_percentage)
name = '{}_{}'.format(dataset_name, split)
__sets[name] = (lambda split=split, dataset_name=dataset_name: water(split,devkit_path=os.path.join('datasets', dataset_name)))
for split in ['train', 'test']:
dataset_name = 'amds'
name = '{}_{}'.format(dataset_name, split)
__sets[name] = (lambda split=split, dataset_name=dataset_name: amds(split,devkit_path=os.path.join('datasets', dataset_name)))
def get_imdb(name):
"""Get an imdb (image database) by name."""
if name not in __sets:
raise KeyError('Unknown dataset: {}'.format(name))
return __sets[name]()
__sets[name] = (lambda split=split: watercolor_car(split, year))
for year in ['2007']:
for split in ['train', 'val']:
name = 'watercolor_{}_{}'.format(year, split)
__sets[name] = (lambda split=split: watercolor(split, year))
for year in ['2007']:
for split in ['train', 'val']:
name = 'clipart_{}_{}'.format(year, split)
__sets[name] = (lambda split=split: clipart(split, year))
for year in ['2007']:
for split in ['train', 'val']:
name = 'comic_{}_{}'.format(year, split)
__sets[name] = (lambda split=split: comic(split, year))
###########################################
def get_imdb(name):
"""Get an imdb (image database) by name."""
if name not in __sets:
raise KeyError('Unknown dataset: {}'.format(name))
return __sets[name]()
def list_imdbs():
"""List all registered imdbs."""
return list(__sets.keys())