Issues building minimal image segmentation example using custom data

Check out some suggestions from here: