search

wurenkai / UltraLight-VM-UNet

[arXiv] The official code for "UltraLight VM-UNet: Parallel Vision Mamba Significantly Reduces Parameters for Skin Lesion Segmentation".
235 stars 33 forks source link

multi-class segmentation #41

Open yhybe opened 6 months ago

yhybe commented 6 months ago

Respected author, I want to inquire about how to apply this to multi-class segmentation. I have used the loss function of VM_mamba, but there were problems. If I directly use the structure of VM_mamba, it is not in the form of slicing.

yhybe commented 6 months ago

could you tell me how to set the loss and metrics scores please!

wurenkai commented 6 months ago

Hi, you can get an answer from this issue.