.
train_datagen.flow_from_directory(directory=train_dir, target_size=(224, 224), class_mode='binary', batch_size=32) test_data = test_datagen.flow_from_directory(directory=test_dir, target_size=(224, 224), class_mode='binary', batch_size=32) Output: Found 1500 images belonging to 2 classes. Found 500