I have started using the tf.data.Dataset as a way to load data into keras models, as they appear to be much faster than keras\' ImageDataGenerator and much more
ImageDataGenerator