Pytorch 中 UNet 的奇怪语义分割结果

编程问答 2022-05-28

问题描述

我只是想用乳房 X 光检查数据集从头开始训练 UNet，以检测乳房 X 光检查中的肿瘤组织。在对模型进行 35 次训练后，我在训练和测试集上得到了奇怪的曲线和糟糕的结果。我在我的代码中找不到任何明显的错误，而且一切似乎都是正确的。

这是在 GitHub 上实现的链接：GitHub - hrnademi/Mammography

关于数据集的更多信息： 训练图像：2400 张 256x256 png 大小的图像 测试图片：600张图片大小为256x256 png

曲线如下：

enter image description here

enter image description here

enter image description here

enter image description here

enter image description here

enter image description here

非常感谢您的关注。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

computer-visionmedical-imagingpytorchsemantic-segmentation