data_loader.py 文件源码

def getImage(datapath, imageID, purpose='train'):
    name_3 = str(imageID)
    name_2 = '0' * (12-len(name_3))
    name_1 = 'COCO_' + purpose + '2014_'
    fileName = name_1 + name_2 + name_3 + '.jpg'
    filepath = join(datapath,fileName)
    img = skimage.io.imread(filepath)
    return(img)