search

JCruan519 / VM-UNet

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

自己的数据集应该如何调整 #27

Open oliea1 opened 7 months ago

oliea1 commented 7 months ago

请问一下哈，我的数据集分为背景，还有另外两种目标，我应该如何设置我的numclass，如何命名数据集呢，数据集如何调整，我把image和mask都设为三通道，运行会报错

oliea1 commented 7 months ago

![Uploading error_pic.PNG…]()

JCruan519 commented 7 months ago

@oliea1 Hello, for binary classification tasks (foreground, background), you can modify the train.py and utils.py files based on the skin disease segmentation task. For multi-class classification tasks (foreground 1, foreground 2, ..., background), you can modify the train_synapse.py and utils_synapse.py files based on the Synapse task.

oliea1 commented 7 months ago

我的图片是jpg，masks是png的

MQN-80 commented 7 months ago

我的图片是jpg，masks是png的

目前看来应该是把自己的数据集按照synapse的格式修改，都用npy.h5格式存放

oliea1 commented 7 months ago

谢谢！

MQN-80 commented 7 months ago

谢谢！ 题主解决了吗，我看他的多分类数据集好像是灰度图片，我是彩色图片，这样需要改模型吗

oliea1 commented 7 months ago

谢谢！ 题主解决了吗，我看他的多分类数据集好像是灰度图片，我是彩色图片，这样需要改模型吗

没有555，我的就是灰度图片，我甚至不知道怎么改

zhaobing17 commented 3 weeks ago

谢谢！

请问解决了吗，我把我的数据集和你的格式一样，只有png和mask，我是有6个类别，但是我把她变成Synapse数据集，运行的时候，一直报找不到我制作的数据集里面的文件，麻烦指教

oliea1 commented 2 days ago

谢谢！

请问解决了吗，我把我的数据集和你的格式一样，只有png和mask，我是有6个类别，但是我把她变成Synapse数据集，运行的时候，一直报找不到我制作的数据集里面的文件，麻烦指教

很遗憾没有