search

liuqk3 / PUT

Paper 'Transformer based Pluralistic Image Completion with Reduced Information Loss' in TPAMI 2024 and 'Reduce Information Loss in Transformers for Pluralistic Image Inpainting' in CVPR2022
MIT License
173 stars 15 forks source link

推理遇到问题 #29

Open superlittl opened 8 months ago

superlittl commented 8 months ago

作者您好，我使用我的训练集进行模型训练一切正常，但在测试的时候遇到了问题。 我的任务和传统的修复有点差别，首先这是我的输入图片： 001 _01 这个是mask： 001 _01 我的意图就是用我的mask能恰好盖住我原始图上的空洞区域，然后进行修复。 但是遇到了问题，发现推理后的图片。并没有识别到我的mask区域，

liuqk3 commented 8 months ago

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