Hi,
I’m working on a problem regarding image segmentation.
My input is RGB, and the labels are 4 channels, one for each of the classes. Technically, the first 3 channels form an RGB image and the last channel is added to differentiate between the other 3 classes and the background.
A typical input and label combo looks like this. This is extracted from the dataloader itself, outside training.

The specific problem is with my labels (which are also images) totally changing somewhere in the process and I’m not quite sure where or how this is happening
During the training process, I printed the label images for debugging purposes, and this is what i get.

Does anyone know whats going on here?
Thanks.