基于色彩聚类的图像修复

作者：冯莹莹; 邱宇; 张登银^*

来源：计算机系统应用, 2023, 32(11): 240-246.

DOI：10.15888/j.cnki.csa.009284

• 图像修复
• 色彩聚类
• 交叉注意力机制
• 渐进式修复

现有图像修复方式普遍存在结构错乱和修复区域边缘模糊的问题,这是由于修复过程中已知区域的结构信息未能得到充分利用,针对这一问题,本文提出了一种具有编解码结构的基于色彩聚类的图像修复算法,算法采用渐进式图像修复网络结构,将图像经过色彩聚类的结果作为输入,聚类算法处理后的图像更好保留了输入图像的结构特征,因此在后续图像纹理恢复过程中结构信息可以得到充分利用;同时,为进一步减少网络的计算开销,引入交叉注意力机制,从水平和垂直两个维度建模图像全局依赖性.实验结果表明,本文提出的图像修复算法可以有效缓解图像修复区域边缘模糊的现象,与几种主流图像修复算法相比,我们提出的图像修复算法可以在缺失区域较大的情况下产生更加真实的输出结果.

• 单位
  通信与信息工程学院; 物联网学院; 南京邮电大学