Unet预测灰度图像(灰色块)

Question

我在unet的实现和我的数据上遇到了问题。unet可以处理测试数据，但不能处理我的数据。我只得到一块灰色像素作为预测。我想把这个unet用在一个大学项目上。有人看到我的错了吗？

这是我试图复制的回购：https://github.com/zhixuhao/unet

这是我的回购：https://github.com/Machuntox/unet

膜数据是测试数据，JOSM数据是我的项目。

你好，斯特凡

Answer 1

我发现了我的错误。我需要对我的图像进行不同的预处理。当我添加alpha通道的值为255时，它起作用了。最后两步做到了这一点。

请参阅提交列表：https://github.com/Machuntox/unet/commits/master/prepareData.py