计算机辅助设计与图形学学报

作者：陈羽中, 林闽沪, 陈友昆, 牛玉贞

关键词：图像雨滴去除, 注意力机制, 多尺度特征, 特征融合

图像雨滴去除任务的目标是对于给定的雨滴图像去除其镜头上遮挡的附着雨滴, 还原出真实的干净图像,其在计算机视觉下游任务中有着至关重要的作用. 由于已有的图像雨滴去除方法没有考虑雨滴所具有的空间位置局部性和尺度多样性, 雨滴去除效果不理想. 为缓解上述问题, 提出一种基于多重注意力机制的图像雨滴去除方法. 首先, 为了适应雨滴的空间位置局部性和尺度多样性, 提出结合多尺度特征提取模块和多重注意力模块构建编码器-解码器架构, 其中多重注意力模块融合了像素、通道和空间注意力, 能够自适应地匹配雨滴的空间位置局部性. 此外,设计了一种迭代式图像特征融合模块, 在融合解码器特征和雨滴图像得到初步去雨图像后, 采用初步去雨图像加强解码器特征, 得到进一步的细化特征, 再融合初步去雨图像和细化特征得到最终去雨图像. 在雨滴图像测试集Raindrop 上实验结果表明, 与其他方法相比, 所提方法能够有效地去除图像中的雨滴, 进一步提升雨滴去除的性能,比对比方法中最优的方法在 PSNR 指标上提升了 0.25 dB.

来源：2025年第5期

《计算机辅助设计与图形学学报》期刊编辑部

查看计算机辅助设计与图形学学报杂志2025年第5期