Hi, I would recommend to save the images as files and use one of ImageDataGenerator methods which you think is best. That being said also look at tf.data