How do I create a Image Segmentation Dataset

🤗Datasets
8

@NimaBoscarino and @nateraw a very naive question, so we have processed the images and now every image represent one single mask, like
Img:
|- Img 1
|- Img 2

Mask:
|- Mask for Img 1
|- Mask for Img 2

and these mask are for single classes, so now I have to create a json file which will have the number of image and then their respective mask. This part I am not understanding at all, as how will the model understand which mask is for which class.