Unet预测灰色图像（灰色块）

Question

我在unet实现和数据方面遇到问题。 Unet适用于testdata，但不适用于我的数据。我只得到一块灰色像素作为预测。我想将此unet用于大学项目。有人看到我的错了吗？

这是我尝试复制的存储库：https://github.com/zhixuhao/unet

这是我的仓库：https://github.com/Machuntox/unet

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

问候， 斯蒂芬

Answer 1

我发现了我的错误。我需要对图像进行不同的预处理。当我添加值为255的Alpha通道时，它起作用了。最后两个步骤可以解决问题。

查看提交列表：https://github.com/Machuntox/unet/commits/master/prepareData.py