可以用于多分类吗

wurenkai / UltraLight-VM-UNet

[arXiv] The official code for "UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation".

223 stars 30 forks source link

可以用于多分类吗 #14

Open yi-coder18 opened 6 months ago

yi-coder18 commented 6 months ago

VM_Unet有用于二分类和多分类的图片的不同训练代码，请问UItraLight-VM-Unet有吗

wurenkai commented 6 months ago

Hi, the multi-targeting task is of course fine. 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!