search

MontaEllis / Pytorch-Medical-Segmentation

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.
MIT License
860 stars 196 forks source link

训练相关 #12

Closed LoinJi closed 3 years ago

LoinJi commented 3 years ago

您好！有一个关于训练结果的问题想要请教一下。 我使用的网上下载的胰腺数据集，原数据和标签都是 nii.gz 格式的，因为是3d数据，所以用的是3d的网络来跑的。是在本地的计算机上跑的代码。默认设置的epoch是100 参数设置如下： image 但是我训练到后面，loss值是在下降，但是dice值就一直是0了，false_positive_rate以及false_negative_rate也很不正常，如下所示： image 想请问原因可以能为什么呢？是否和epoch数有关？因为我看您的原代码是500000，我设置100已经需要跑一段时间了。 其他的参数是否有要根据数据的三维尺寸调整的呢？ 感谢回复！

LoinJi commented 3 years ago

会不会和数据量小有关呢？因为还在初期阶段，所以只用了10个病例来试试。而且最后试着预测了一下，输出的结果看起来是由几个大方块组成的，类似于像素块那种。

我训练期间输出dice，false_positive_rate以及false_negative_rate的代码如下： image

MontaEllis commented 3 years ago

请把你的训练log给我看一下

LoinJi commented 3 years ago

image 是这些文件吗？

MontaEllis commented 3 years ago

tensorboard的

LoinJi commented 3 years ago

我是在本地跑的，所以没有做tensorboard可视化

LoinJi commented 3 years ago

我刚才可视化了一下，结果是这样的。 image 我现在还在跑，这个是用30个病例跑的

LoinJi commented 3 years ago

这是我现在用的数据和标签 image

MontaEllis commented 3 years ago

你的label是0-1吗？用的是最新的代码吗？

LoinJi commented 3 years ago

是最新的代码，标签是这样的 image 应该是0-1的吧

MontaEllis commented 3 years ago

mitk可以看那个值的，点着看一下。你可以加我微信聊下

LoinJi commented 3 years ago

好的，谢谢

daisysong1 commented 2 years ago

mitk可以看那个值的，点着看一下。你可以加我微信聊下

我也有问题想请教，如何加您的微信呀