Preprocess images

datapreprocessing.py

def datapreprocessing(dataframe,bsize):
    from tensorflow.keras.preprocessing.image import ImageDataGenerator
    
    train_gen = ImageDataGenerator(zoom_range=0.5,
                                   rescale=1.0/255,
                                   horizontal_flip=True,
                                   rotation_range=40,)

    train_generator = train_gen.flow_from_dataframe(
        dataframe,
        x_col= dataframe.columns[0],
        y_col=dataframe.columns[1],
        target_size=(150,150),
        batch_size=bsize,
        color_mode="rgb",
        shuffle=True,
        class_mode='categorical')
    
    return train_generator