I am using tensorflow.keras.preprocessing.image.ImageDataGenerator to create a iterable dataset. Below is my code for it.
datagen = ImageDataGenerator( rescal