search

MontaEllis / Pytorch-Medical-Segmentation

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

我的损失函数从正值一直减小到负数了？并且损失函数的绝对值还在不断增大 #32

Closed Fuzhiyuan closed 2 years ago

Fuzhiyuan commented 2 years ago

这该怎么办？

Fuzhiyuan commented 2 years ago

学习率也没有更改，一直是0.002

JohnMasoner commented 2 years ago

数据格式是dicom格式的吗? 你可以将'main.py'文件中'#print(y.max())'注释掉观察一下 label 的最大值.确保 label 为1.

Fuzhiyuan commented 2 years ago

数据格式是dicom格式的吗? 你可以将'main.py'文件中'#print(y.max())'注释掉观察一下 label 的最大值.确保 label 为1.

数据格式为.nii.gz 我的模型是3d unet模型

Fuzhiyuan commented 2 years ago

数据格式是dicom格式的吗? 你可以将'main.py'文件中'#print(y.max())'注释掉观察一下 label 的最大值.确保 label 为1.

还有就是label最大值为1是什么意思啊？不是很明白。

JohnMasoner commented 2 years ago

label 的标注格式可以是[0,255], 也可以是 [0,1]. 这个模型作者写的模型仅仅适用0,1格式的标注. 如果无法理解, 可以将'main.py'的'#y[y!=0] = 1'注释取消掉来保证你的label格式为0,1.

Fuzhiyuan commented 2 years ago

label 的标注格式可以是[0,255], 也可以是 [0,1]. 这个模型作者写的模型仅仅适用0,1格式的标注. 如果无法理解, 可以将'main.py'的'#y[y!=0] = 1'注释取消掉来保证你的label格式为0,1.

你好，我想问一下如果我做了这个转换，对我的后续训练会有什么影响吗？如果可以的话，可以留一下国内社交软件联系方式吗？邮箱也可以。

JohnMasoner commented 2 years ago

对后续没有影响的, 这一步只是将 label 的标签转为0,1格式,以符合模型对数据的需求. 你可以添加我的微信: MasonMa-