How do I create a Image Segmentation Dataset

Using this you can create a dataset class and read and create a dataset dict, this format worked for me along with the model.