Code for blog article https://yann-leguilly.gitlab.io/post/2019-12-14-tensorflow-tfdata-segmentation/ - View it on GitHub