search

McGregorWwww / UDTransNet

This repo is the official implementation of 'Narrowing the semantic gaps in U-Net with learnable skip connections: The case of medical image segmentation' which is an improved journal version of UCTransNet.
MIT License
74 stars 8 forks source link

多类别分割的test #6

Closed ManuJuice closed 1 year ago

ManuJuice commented 1 year ago

您好，昨天在您的指导下，我完成了多类别分割的训练，验证集的精度也相当可观，但是在运行test_kfold.py时，遇到了一些问题。

首先，我是基于GlaS数据集的格式创建了我的数据集，并且将task_name设为GlaS进行训练，得到了最终的best_model；

接着，我将Config.py里的test_session改成对应的任务，并运行test_kfold.py，经过我的一系列调整，仍在计算卷积时报错： File "C:\Users\hp\Desktop\ud\nets\UDTransNet.py", line 199, in forward e1 = self.Conv1(x)

报错如下： RuntimeError: Given groups=1, weight of size [64, 3, 3, 3], expected input[1, 1, 224, 224] to have 3 channels, but got 1 channels instead

鉴于task_name为GlaS时，test是针对二分类分割任务的，我想向您请教一下，对于多类别分割任务，在test时需要对那些部分做修改？

McGregorWwww commented 1 year ago

你好，你这个错误应该是读取图片的问题，GlaS数据集的图片是RGB三通道的，你的应该是1通道灰度图，改一下config里的channel数量就好了

ManuJuice commented 1 year ago

你好，你这个错误应该是读取图片的问题，GlaS数据集的图片是RGB三通道的，你的应该是1通道灰度图，改一下config里的channel数量就好了

谢谢，已经解决了。不过channel没有错，是读取数据后的处理导致最后输入模型的图片不对

yese12 commented 7 months ago

你好，我想问下在验证时，出现FileNotFoundError: [WinError 3] 系统找不到指定的路径。: './datasets/Synapse/test_vol_h5//img' 该怎么解决，目前我的文件路径是没有问题的，test_vol_h5里原数据集存放的是npy.h5文件，是因为格式不对导致加载不出来吗，期待您的回复

watchingyoukk commented 3 weeks ago

你好，你这个错误应该是读取图片的问题，GlaS数据集的图片是RGB三通道的，你的应该是1通道灰度图，改一下config里的channel数量就好了

谢谢，已经解决了。不过channel没有错，是读取数据后的处理导致最后输入模型的图片不对

你好方便问一下修改了哪些可以测试多类的glas的代码吗