keras_image_generator.py

from keras.utils import Sequence
class DataSequence(Sequence):
    """
    Keras Sequence object to train a model on larger-than-memory data.
    """
    def __init__(self, df, batch_size, mode='train'):
      
    ...
    
    def get_batch_images(self, idx):
        # Read a batch of images from disk
        return np.array([imread(im) for im in self.im_list[idx * self.bsz: (1 + idx) * self.bsz]])

    def __getitem__(self, idx):
        batch_x = self.get_batch_images(idx)
        batch_y = self.get_batch_labels(idx)
        return batch_x, batch_y