多尺度编码-解码结构的单幅图像去雨网络

作者：张学锋; 李金晶; 储岳中; 汤亚玲

来源：华中科技大学学报(自然科学版)科技大学, 2021, 49(07): 68-73.

DOI：10.13245/j.hust.210713

• 单幅图像去雨
• 扩张卷积
• 双路网络
• 通道注意力
• 长短时间记忆网络

针对图像去雨问题提出多尺度编码-解码结构的单幅图像去雨网络(MSEDNet).首先在编码阶段对雨图像进行二次下采样;然后每个下采样层分别通过特征细化模块提取特征信息;最后将特征图逐步上采样,并且与上一个阶段的特征图聚合得到最终的去雨图像.采用不同分辨率的特征提取网络,能更好地提取雨纹的特征.特征细化网络首先经过多尺度扩张卷积聚合雨纹特征,然后通道注意力给聚合模块通道分配不同的权重以细化自身,最后使用ResNet和DenseNet的双路特征提取器实现雨纹信息提取.仿真和真实雨数据集结果表明:所提网络能够有效地去除雨纹,得到高质量的无雨图像.

• 单位
  安徽工业大学