一种结合自注意和多尺度生成对抗网络的图像去雨方法

来源：计算机工程与科学杂志2021年第12期北京时间：

作者：李然, 周子淏, 张月芳, 罗东升, 邓红霞

单位：太原理工大学信息与计算机学院，山西 晋中 030600

摘要：为去除雨天拍摄照片上的雨滴，针对被雨滴所覆盖区域未知，雨滴区域中大多数背景信息已经丢失，以及需要提升图像清晰度和对全局信息关注度的问题，在生成对抗网络中生成网络的自动编码器结构中添加自注意层，并在判别网络中引入多尺度判别器。通过注意力分布图的引导，自注意层的优化和多尺度判别器的评估，生成网络在关注雨滴区域的前提下进一步关注全局信息，多尺度判别器可由粗到细更好地判别雨滴图像与清晰图像之间的差距。实验完成了所提方法与其他方法的对比，以及自对比，并用峰值信噪比和结构相似性进行评估，结果表明了所提方法的有效性，其质量和指标数值均高于其他方法。

关键词：去雨,生成对抗网络,自我注意,多尺度,

基金资助：山西省重点研发计划(201803D31038);晋中市科技重点研发项目(Y192006);山西省自然科学基金(201801D121135)