coco_loader.py 文件源码

def __init__(self, split, memoization=False):
        assert not memoization
        self.dataset = 'coco'
        self.coco_ids_to_internal = {k: self.cats_to_ids[v] for k, v in coco_ids_to_cats.items()}
        self.ids_to_coco_ids = dict(map(reversed, self.coco_ids_to_internal.items()))
        self.split = split
        assert self.split in ['train2014', 'val2014', 'test2014', 'test2015', 'minival2014', 'valminusminival2014', 'test-dev2015']
        self.root = DATASETS_ROOT + 'coco/'
        self.included_coco_ids = list(coco_ids.values())
        if 'test' in self.split:
            json = '%s/annotations/image_info_%s.json'
        else:
            json = '%s/annotations/instances_%s.json'
        self.coco = COCO(json % (self.root, self.split))
        self.filenames = self.coco.getImgIds()
        self.num_classes = COCOLoader.num_classes
        self.real_split = self.split
        if self.real_split in ['minival2014', 'valminusminival2014']:
            self.real_split = 'val2014'
            self.coco = COCO('%s/annotations/instances_%s.json' % (self.root, self.real_split))
        if 'test' in self.real_split:
            self.real_split = 'test2015'
        log.info("Created a COCO loader %s with %i images" % (split, len(self.coco.getImgIds())))