为什么预测出来的图片全黑，能不能回复一下

milesial / Pytorch-UNet

PyTorch implementation of the U-Net for image semantic segmentation with high quality images

GNU General Public License v3.0

9.3k stars 2.51k forks source link

Closed FRC1221 closed 8 months ago

huyingjiao commented 8 months ago

你好，我遇到了相同的问题，你的问题解决了吗

FRC1221 commented 8 months ago

你好，我遇到了相同的问题，你的问题解决了吗

没有，换其他的

HunterTom94 commented 7 months ago

你是不是跑train的时候用了 --amp？可以试试不要用amp再train一遍

deeperlearner commented 5 months ago

I have the same problem. I did not use amp. Did you solve it?

deeperlearner commented 5 months ago

In my case, maybe smaller batch-size can solve it.