JCruan519/VM-UNet

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

oliea1 opened this issue · 7 comments

oliea1 commented

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

JCruan519 commented

@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

我的图片是jpg，masks是png的

MQN-80 commented

我的图片是jpg，masks是png的

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

oliea1 commented

谢谢！

MQN-80 commented

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

oliea1 commented

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

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