基于完全残差的双分支去雨网络

来源：计算机技术与发展杂志2023年第11期北京时间：

作者：宋浩;'*;张鸿;'

单位：1. 武汉科技大学 计算机科学与技术学院,湖北 武汉 430065;2. 智能信息处理与实时工业系统湖北省重点实验室(武汉科技大学),湖北 武汉 430065

摘要：针对使用深度学习的单幅图像去雨算法会导致细节信息丢失的问题,提出了一个双分支去雨网络,包括雨痕提取分支和细节恢复分支,通过补全细节使去雨图像更接近真实图像。 雨痕提取分支的目的是完全提取出雨纹,通过构造特征金字塔来多尺度地学习雨纹信息,并在其中引入执行了全部身份映射的完全残差块来加强特征的重用和传播。 为充分利用上下文信息,采用可变形卷积在动态扩大感受野的同时避免了网格伪影的产生,最后输入雨图去除雨痕便得到了初步去雨图。 细节恢复分支需要产生细节特征图反馈给初步去雨图像来找回丢失的细节,使用轻量级的完全残差块捕捉特征信息,并用跳跃连接来连接完全残差块提供长距离的信息补偿。 实验结果表明,该网络在合成数据集 Rain100H 中比较RESCAN、SPANet 和 JDNet 等主流去雨方法,在 PSNR 和 SSIM 指标上分别至少提高了 0. 09 dB 和 0. 02,在真实数据集和自制数据集中的去雨效果和细节保留程度均优于对比方法。

关键词：卷积神经网络;单幅图像去雨;多尺度学习;完全残差;可变形卷积