2D multi-class classification tasks

[cdzheng2g]

I want to ask how to adapt it for 2D multi-class classification tasks. What do I need to modify?

[wurenkai]

Hi, You can refer to issue 14. Specifically, you just need to replace 'vmunet.py' in the model folder with 'UItraLight-VM-UNet.py' with the help of VM-UNet's multi-targeting framework (train_synapse.py) '. Also, and modify the output 'return torch.sigmoid(out0)' to 'return out0' to run the multi-target task!