create_keras_data_generators.py

# create training and validation data generators


IMG_SIZE = (224,224)
BATCH_SIZE = 8


def train_val_generators(train_dir, val_dir):
   train_gen = ImageDataGenerator(rescale=1/255.,
                                 horizontal_flip=True,
                                 vertical_flip=True,
                                 brightness_range=(0, 0.2),
                                 rotation_range=0.2,
                                 zoom_range=0.2,)
  
   val_gen = ImageDataGenerator(rescale=1/255.)

   
 

# create image generators to feed images to the model
train_generator, val_generator = train_val_generators(train_dir, val_dir) 

   train_generator = train_gen.flow_from_directory(directory=train_dir,
                                                 batch_size=BATCH_SIZE,
                                                 class_mode="categorical",
                                                 target_size=IMG_SIZE)
  
   val_generator = val_gen.flow_from_directory(directory=val_dir,
                                             batch_size=BATCH_SIZE,
                                             class_mode="categorical",
                                             target_size=IMG_SIZE)
   return train_generator, val_generator