name = 'voc_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: pascal_voc(split, year))
# Set up voc_water_<year>_<split>
for year in ['2007', '2012']:
for split in ['train', 'val', 'trainval', 'test']:
name = 'voc_water_{}_{}'.format(year, split)
__sets[name] = (
lambda split=split, year=year: pascal_voc_water(split, year))
# Set up clipart_<year>_<split>
for year in ['2007']:
devkit_path = '/userhome/Datasets/clipart'
for split in ['train', 'test', 'trainval']:
name = 'clipart_{}_{}'.format(year, split)
__sets[name] = (lambda year=year, split=split, devkit_path=devkit_path:
clipart(split, year, devkit_path=devkit_path))
# Set up dt_clipart voc_<year>_<split>
for year in ['2007', '2012']:
devkit_path = '/userhome/Datasets/dt_clipart'
for split in ['train', 'val', 'trainval', 'test']:
name = 'dt_clipart_voc_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year, devkit_path=devkit_path:
dt_clipart(split, year, devkit_path=devkit_path))
# Set up clipart2VOC_<year>_<split>
for year in ['2007']:
devkit_path = '/userhome/Datasets/clipart2VOC'
for split in ['train', 'test', 'trainval']:
name = 'clipart2VOC_{}_{}'.format(year, split)
__sets[name] = (lambda year=year, split=split, devkit_path=devkit_path:
clipart2VOC(split, year, devkit_path=devkit_path))
# Set up pl_clipart voc_<year>_<split>
__sets[name] = (
lambda split=split, year=year: pascal_voc_water(split, year))
for year in ['2007', '2012']:
for split in ['train', 'val', 'trainval', 'test']:
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 year in ['2007']:
for split in ['trainval', 'test']:
name = 'clipart_{}'.format(split)
__sets[name] = (lambda split=split: clipart(split, year))
for year in ['2007']:
for split in ['train', 'test']:
name = 'water_{}'.format(split)
__sets[name] = (lambda split=split: water(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."""
for split in ['train', 'val', 'trainval', 'test']:
name = 'voc_water_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: pascal_voc_water(split, year))
for year in ['2007', '2012']:
for split in ['train', 'val', 'trainval', 'test']:
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):
data_percentage_split: voc_clipart(
'clipart{}'.format(data_percentage_split),
shift,
split,
devkit_path=os.path.join(
'datasets/', 'clipart{}_{}'.format(
data_percentage_split, shift))))
# 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,
from datasets.comic import comic
import numpy as np
# Set up voc_<year>_<split>
for year in ['2007', '2012']:
for split in ['train', 'val', 'trainval', 'test', 'trainvalsemi']:
name = 'voc_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: pascal_voc(split, year))
# 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)
