推理遇到问题

[superlittl]

作者您好，我使用我的训练集进行模型训练一切正常，但在测试的时候遇到了问题。
我的任务和传统的修复有点差别，首先这是我的输入图片：

这个是mask：

我的意图就是用我的mask能恰好盖住我原始图上的空洞区域，然后进行修复。
但是遇到了问题，发现推理后的图片。并没有识别到我的mask区域，

[liuqk3]

感谢关注。代码中根据提供的mask，将白色区域（像素值为255）作为unmask区域，黑色区域（0像素值为0）作为mask区域。你这边的mask应该刚好反过来了。