U-Net: A PyTorch Implementation in 60 lines of Code