Custom dataset
You could also create and use your own dataset if you prefer to train with the run_semantic_segmentation.py script instead of a notebook instance. The script requires:

a [~datasets.DatasetDict] with two [~datasets.Image] columns, "image" and "label"