https://analyticsindiamag.com/my-experiment-with-unet-building-an-image-segmentation-model/