Multiclass segmentation with different labels and satellite unet

[shubham-scisar]

I just have a problem in which I have images out of which some have 2 classes and some have 3 classes. So, basically the 2 classes are already there in all images but some have additional one class. Now, while dealing with multiclass segmentation ,I have to make the size (width,height,classes) so in this case it will cause inconsistency. As all the images should have same dimensions. Kindly guide.

Further, I have to apply multiclass segmentation for satellite unet specifically and not use custom unet for that. As specified in the docs, that custom unet supports multiclass segementation. Kindly guide.