https://learnmodernpython.com/tensorflow-building-a-simple-image-segmentation-model/