I want to be able to work with datasets of paired image training examples (like a source image and segmentation map) and add geometric transforms into my pipeline that I can