注意力与多尺度融合的图像阴影去除算法

作者：曲海成; 佟畅; 刘万军

来源：计算机工程与应用, 2021, 1-11.

• 注意力机制
• 多尺度特征融合
• 生成对抗网络
• 阴影去除

针对图像阴影去除算法中复杂地物或与阴影区域纹理相似的暗区域阴影去除不完全的问题，提出了一种注意力与多尺度融合的图像阴影去除算法。该算法基于生成对抗网络框架构建。首先，利用自定义的空洞残差块进行特征提取，获得精确的阴影特征信息并输入到注意力引导的编码网络；其次，在注意力引导的编码阶段进行多尺度的特征融合，获取不同层次的感受野，使编码器能够在不同尺度上捕捉上下文信息；最后，利用多重注意力机制引导判别网络对生成的无阴影图像进行鉴别，进而减少判别网络关键信息损失，提高判别网络的鉴别效果。该算法分别在ISTD与SRD公开数据集上进行验证，实验结果表明：该算法视觉效果表现良好，单幅阴影去除后的图片与数据集中真实无阴影图片进行对比，SSIM可达到97.8%，PSNR可达到32.2dB，RMSE可达到6.2。相比同类算法，具有显著优势，且对复杂地物或暗区域阴影去除效果良好。

• 单位
  辽宁工程技术大学