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):