Need help with dataset

https://www.tensorflow.org/tutorials/images/segmentation

This tutorial is exactly what I am looking for. Rather than multiple types of ‘animals’ I have two ‘species’ that I want segmented. The predicted mask as the outcome is what I want. I have my data right now, my input image and true masks, but I’m really confused about formatting my dataset similar to this one so I can use the same model for my project.

I am very new to TensorFlow, so any help would be appreciated.

Thank y’all so much!