i have a mask file saved as a .tiff file ranging from 0…255.
I would like to split it into X classes so my tensor look something like this [batch, classes, width, height].
My network currently does Binary classification and outputs [9, 2, 256, 256].
So i would like to match this with the segmentation mask?, and understand how i should do if i wannt to increase the number of classes?