change code for Multi-classification problem

[Little1salt2fish]

hello If I'd like to use the model to classify more than two categories, could you give me some advise about change which code？

[JuliaWolleb]

Hi Sure, you need to have a one-hot-encoding for the multiple classes. So you will have one binary output channel per class. The only thing you need to change is the number of channels in the training dataset, and the number of input and output channel of your U-Net architecture. The rest should work the same.

Let me know if you have further questions.