text_plain_output_1.png
from keras.preprocessing.image import ImageDataGenerator

train_data_dir = '../input/1056lab-covid19-chest-xray-recognit/train'
generator = ImageDataGenerator(width_shift_range=0.3, height_shift_range=0.3, horizontal_flip=True, validation_split=0.2)

train_generator = generator.flow_from_directory(train_data_dir, target_size=(224, 224), color_mode='rgb', batch_size=64, class_mode='categorical', shuffle=True)