U Net Image Segmentation Pytorch Dataloader