pytorch图像分割



我一直在尝试通过使用Esertel等人发布的作品来实现图像分割模型,2022,代码可在这里获得(https://github.com/burakekim/LULCMapping-WV3images-CORINE-DLMethods/blob/main/lulc_wv3.ipynb)。

但是我发现很难将它用于主数据(尽管,由于PC RAM较小,我使用了较小的图像(32X32像素))。在一次健全检查中,我发现所有的掩模都是黑色的,训练后掩模仍然是黑色的,预测是有偏差的(见图),这可能会导致下一个问题,这个问题与非常高的指标相关,例如验证:100%,精度-1.0,召回率1.0等。

输入图片描述

由于我是人工智能的初学者,因此可能这个问题也没有得到足够的描述,无论如何也许有人可以帮助我解决这个问题。

找到了非常简单的解决方案,我没有在调色板中指定主要的RGB颜色。

最新更新