预测出现全黑问题

Open wzr0108 opened this issue 3 years ago • 5 comments

用voc训练集训练出来的模型，预测时全黑 `# ---------------------------------------------------#

取出每一个像素点的种类

pr = pr.argmax(axis=-1)`

这个pr是全0的，用预训练模型继续训练，也会出现黑点越来越多的情况，请问这是什么原因呢

Feb 24 '22 13:02 wzr0108

嗯？看一下注释，标签有问题

Feb 25 '22 03:02 bubbliiiing

标签应该没问题啊，我看了DeeplabDataset返回的png 屏幕截图 2022-02-25 112039 这是每一类对应的像素点的个数，是正常的吧

Feb 25 '22 03:02 wzr0108

我也遇到了这个问题，训练是正常的，预测突然就全是黑的了

Mar 25 '22 10:03 zysc1996