dataset = keras.preprocessing.image_dataset_from_directory( 'path/to/main_directory', batch_size=64, image_size=(200, 200))
gen = keras.preprocessing.image.ImageDataGenerator()
dataset = keras.preprocessing.image_dataset_from_directory( 'path/to/main_directory', batch_size=64, image_size=(200, 200))
gen = keras.preprocessing.image.ImageDataGenerator()