text_plain_output_1.png
from keras.preprocessing.image import ImageDataGenerator

train_datagen = ImageDataGenerator(rescale=1.0 / 255.0)
test_datagen = ImageDataGenerator(rescale=1.0 / 255.0)
train_it = train_datagen.flow_from_directory('/kaggle/input/dataset/train/', class_mode='categorical', batch_size=10, target_size=(224, 224))
test_it = test_datagen.flow_from_directory('/kaggle/input/dataset/test/', class_mode='categorical', batch_size=5, target_size=(224, 224))

train_it.class_indices