Thank you very much for the reply. I have 2 follow up questions -
1 - in binary, the output of the U-net model will is a softmax(). Will it be the same for multi-class?
2 - my target has 3 classes. So, I should have 0 for background and 1,2,3 for the three classes at each pixel location, right?