Imported from https://github.com/ugent-korea/pytorch-unet-segmentation