U-Net for image segmentation, PyTorch implementation. - View it on GitHub