MontaEllis / Pytorch-Medical-Segmentation

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

main.py第250行loss = criterion(outputs, y)-->loss = criterion(logits, y) #33

Closed qianjinfighter closed 2 years ago

qianjinfighter commented 2 years ago

你好,很感谢您的代码,我预测效果很差,多次修改损失函数都没有效果,看代码发现这里有点疑问,请问这里是不是错了? man.py代码: logits = torch.sigmoid(outputs) labels = logits.clone() labels[labels>0.5] = 1 labels[labels<=0.5] = 0 loss = criterion(outputs, y)#有问题?应该为loss = criterion(logits, y)

MontaEllis commented 2 years ago

没有问题

qianjinfighter commented 2 years ago

啊 好的 谢谢