dataIO.py 文件源码

def __init__(self, datadir,
                 target_size = 108, image_size = 64,
                 split = 5, num_utilize = np.inf):

        self.datadir = datadir
        self.target_size = target_size
        self.image_size = image_size
        self.split = split

        self.image_paths = []
        for d in self.datadir:
            self.image_paths += glob(d + '/*.jpg')
        self.data_size = min(len(self.image_paths), num_utilize)
        print('data size : {}'.format(self.data_size))
        self.image_paths = np.random.choice(self.image_paths,
                                            self.data_size,
                                            replace = False)
        self.data = None