JCruan519 / VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".
Apache License 2.0
393 stars 12 forks source link

多分类 #55

Open xiaogege1210 opened 1 month ago

xiaogege1210 commented 1 month ago

你好,我想问一下你试过用vmunet去跑多分类的数据集吗,我如果想跑多分类的代码我应该怎么修改你的代码呢

JCruan519 commented 1 month ago

@xiaogege1210 Hello, for multi-class classification tasks, you can refer to the Synapse dataset.