search

wjzhang-ty / ExerciseUNet

UNet by torch1.11 for Optic Disc and Cup Segmentation
6 stars 0 forks source link

unet网络结构 #2

Open kkkjane19 opened 1 year ago

kkkjane19 commented 1 year ago

你好小白求问您的unet结构是原unet结构代码吗，有改参数或者加入padding之类的吗，目前代码还不太熟练求教

wjzhang-ty commented 1 year ago

结构是一致的，会有些细节上的不同，但没啥影响，这个网络太老了。 https://github.com/milesial/Pytorch-UNet 这是个很好的学习案例

kkkjane19 commented 1 year ago

结果是一致的，会有一些细节上的不同，但是没有啥影响，这个网络太老了。 https://github.com/milesial/Pytorch-UNet 这是一个很好的学习案例 因为看到您的代码有注释对新手很友好哈哈非常感谢，最后求问下您的这个大概需要训练多少次呢，我多次更改了epochs结果还没有收敛

wjzhang-ty commented 1 year ago

额 我依旧强烈建议你别看这个项目。 这个项目是在StackOverflow上提问的demo。（评估网络时，调整batchsize得到的分数会变化）。我并没有花时间在模型的其他方面。

另外，如果遇到模型无法收敛，你可以尝试跟踪下损失，在图像处理和损失计算这两块出错的可能性较大

kkkjane19 commented 1 year ago

好的谢谢您的耐心解答！