预测出来的图片是灰的

[2361098148]

按照b站的教学来的，训练的自己的图片，为什么预测出来的照片，就像是加了一层灰色蒙版似的，没有识别出来？

[bubbliiiing]

是不是标签格式不对，voc数据集格式，然后你是分两类?

[2361098148]

格式是按照b站来的，自己做的vocal数据集，我是只识别楼梯，所以只分一类算是，这样不可以么？之前我尝试了下说标签必须是黑白的，我改了之后也是没用。

[bubbliiiing]

emm怎么说呢。 要按照voc数据集的格式。 标签的每一个像素点的值为这个像素点的种类。 你是只识别楼梯，也就是背景的像素点值为0，楼梯像素点值为1

[bubbliiiing]

你看看对不对

[2361098148]

你看看对不对

非常感谢您的回复，已经找到问题所在了，标签按照您的视频来做没问题，最后发现是train的参数设置不太合适，更改了lr值为1e-3就可以输出结果了。

[bubbliiiing]

0 0咦，学习要设置这么高吗。我这里一般都是1e-4诶。你训练了多久啊

[2361098148]

我是只训练100张图片，训练了50次，因为不知道为啥超过50次就会显示out of memory！

[Liuhao-128]

我训练完，运行Predict文件，然后不出分割结果。最后我保存了分割结果，打开是全白的

[bubbliiiing]

out of memory batch太大 不出分割结果考虑数据集格式问题