datasets.py 文件源码

def __init__(self, root, annFile, transform=None, target_transform=None):

        from pycocotools.coco import COCO

        self.root = root
        self.coco = COCO(annFile)
        self.ids = list(self.coco.imgs.keys())
        self.transform = transform
        self.target_transform = target_transform