keras_unet.utils.get_augmented read from disk

Question

I want to train a cnn model on a large dataset (10K images and masks).

Currently, I am reading data in batches of BATCH_SIZE = 500 images, augment with