train_dir = 'path/to/train/directory' validation_dir = 'path/to/validation/directory' batch_size = 32 target_size = (227, 227) class_mode = 'binary'

Create train data generator

train_datagen = image.ImageDataGenerator(rescale=1. / 255) train_generator = train_datagen.flow_from_directory( train_dir, target_size=target_size, batch_size=batch_size, class_mode=class_mode )

Create validation data generator

val_datagen = image.ImageDataGenerator(rescale=1. / 255) validation_generator = val_datagen.flow_from_directory( validation_dir, target_size=target_size, batch_size=batch_size, class_mode=class_mode )