multi-class segmentation discuss

[smallboy-code]

Hi @JuliaWolleb , thank you for your great work. I want to solve the multi-class segmentation problem and I delete the "label = torch.where(label > 0, 1, 0).float()". Now, I got some output looks not well as follows:

So, can you give me some advices about this problem?

[JuliaWolleb]

Hi
How did you train this multi-class approach? It seems like you trained on a segmentation mask with a single channel, but different values (0,1,2,3) for the different classes?
I suggest you train with one-hot-encoding, i.e., one binary mask for each of the classes.

[smallboy-code]

Yes, I trained on a segmentation mask with a single channel with different values (0,1,2,3) for the different classes. The one binary mask for each of the classes means training on a segmentation mask with three channels with WT,TC and ET?

[JuliaWolleb]

yes exactly, all three classes will have a separate binary channel each.

[smallboy-code]

Yes, I have also try to do this, but I found the finnal output will be like this:

WT,TC,ET,respectively. I think the results are worse than one channel.

[ZhengChen6]

Yes, I have also try to do this, but I found the finnal output will be like this: WT,TC,ET,respectively. I think the results are worse than one channel.

Hi, I also want to use multi-class segmentation for training, have you solved this problem? @smallboy-code

[smallboy-code]

It remains unresolved. @ZhengChen6