image_generator.py

# importing the library
import tensorflow as tf

# generating new images
train_datagen = tf.keras.preprocessing.image.ImageDataGenerator(rescale = 1./255,
                                                                rotation_range = 60,
                                                                width_shift_range = 0.2,
                                                                height_shift_range = 0.2,
                                                                shear_range = 0.2,
                                                                zoom_range = 0.2,
                                                                horizontal_flip = True)

# storing the images in batches
train_generator = train_datagen.flow_from_directory(training_dir,
                                                    target_size = (300, 300),
                                                    class_mode = 'binary',
                                                    batch_size = 128)

# rescaling the px values
validation_datagen = tf.keras.preprocessing.image.ImageDataGenerator(rescale = 1./255)

# storing the images in batches
validation_generator = validation_datagen.flow_from_directory(validation_dir,
                                                              target_size = (300, 300),
                                                              class_mode = 'binary',
                                                              batch_size = 128)