Problem with segmentation masks

I’m having trouble with segmentation masks, I saved my masks as greyscale images with different values for different classes. for example, I have a value of 1 for class one pixel, till class 4 and the rest of the pixels are zeros. When I check my data bunch, it does not read these masks properly for some reason, I only get zeros for all the pixels. I have cross-checked the saved images and they do contain those pixels properly. Any suggestions?

1 Like

So when you use open_mask are you seeing the expected mask image ?

1 Like

Yes, it turns out that the transforms were the problem. After removing the transforms it worked properly.

You probably don’t want to remove transforms as that is removing data augmentation. Not sure what is happening. Do you have tfm_y = True ?