Unet 预测灰度图像（灰块）

Question

我在使用 unet 实现和我的数据时遇到问题。 unet 适用于测试数据，但不适用于我的数据。我只得到一块灰色像素作为预测。我想将此 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