I have a large dataset of images stored in TFRecord files, and I want to train a neural network on this dataset. My goal is to apply custom augmentations to the images before feeding them into the model. However, I couldn’t find a built-in TensorFlow function like ImageDataGenerator to apply augmentations directly to images stored as tensors before training.