I am trying to perform image segmentation using U-Net. I used tfds oxford pet dataset. Now I want to do the same for the new images I have. my input as well as output is als