Keras data augmentation pipeline for image segmentation dataset (image and mask with same manipulation)

Question

I am building a preprocessing and data augmentation pipeline for my image segmentation dataset There is a powerful API from keras to do this but I ran into the problem of re