Clone and modify based on https://github.com/divamgupta/image-segmentation-keras - View it on GitHub