from lib.datasets.DIY_pascal_voc import DIY_pascal_voc
from lib.datasets.dist_fake import dist_fake
import numpy as np
# Set up voc_<year>_<split>
for year in ['2007', '2012']:
for split in ['train', 'val', 'trainval', 'test']:
name = 'voc_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: pascal_voc(split, year))
# 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_2015_<split>
for year in ['2015']:
for split in ['test', 'test-dev']:
name = 'coco_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: coco(split, year))
for year in ['2007']:
for split in ['trainval']:
name = 'DIY_dataset'
__sets[name] = (
lambda split=split, year=year: DIY_pascal_voc(split, year))
coco_path = '/csai02/scyew1/RGB-N/coco_synthetic/'
for split in ['coco_train_filter', 'coco_test_filter']:
from lib.datasets.kitti_voc import kitti_voc
import numpy as np
# Set up voc_<year>_<split>
for year in ['2007', '2012']:
for split in ['train', 'val', 'trainval', 'test']:
name = 'voc_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: pascal_voc(split, year))
# lambda作为一个表达式，定义了一个匿名函数, :前是函数输入 之后是输出 函数名师 = 前面的_sets
# 很明显，pascal_voc是一个类，这是调用了该类的构造函数，返回的也是该类的一个实例，所以这下我们清楚了imdb实际上就是pascal_voc的一个实例；
# 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_2015_<split>
for year in ['2015']:
for split in ['test', 'test-dev']:
name = 'coco_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: coco(split, year))
# Set up kitti_2012_<split>
for year in ['2012']:
for split in ['train', 'val', 'trainval', 'test']:
name = 'kitti_{}_{}'.format(year, split)
__sets[name] = (lambda split=split, year=year: kitti_voc(split, year))
def get_imdb(name):