Unet预测灰度图像(灰块)

Unet predicts grey images (grey block)

我在 unet 实施和我的数据方面遇到问题。 unet 适用于测试数据,但不适用于我的数据。我只得到一块灰色像素作为预测。我想将这个 unet 用于大学项目。有人看出我的错了吗?

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

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

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

问候, 斯蒂芬

我发现了我的错误。我需要以不同的方式预处理我的图像。当我添加值为 255 的 alpha 通道时,它起作用了。最后两步成功了。

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