import tensorflow as tf

ds_train, ds_valid = tf.keras.preprocessing.image_dataset_from_directory(
    '../dataset-orig',
    labels = 'inferred',
    label_mode = 'categorical',
    color_mode = 'rgb',
    image_size = (300, 300),
    shuffle = False,
    validation_split = 0.05,
    subset = 'both',
    verbose = True
)

from m import mod