多尺度编码-解码结构的单幅图像去雨网络 |
| |
作者单位: | 安徽工业大学计算机与科学学院,安徽马鞍山 243032 |
| |
摘 要: | 针对图像去雨问题提出多尺度编码-解码结构的单幅图像去雨网络(MSEDNet).首先在编码阶段对雨图像进行二次下采样;然后每个下采样层分别通过特征细化模块提取特征信息;最后将特征图逐步上采样,并且与上一个阶段的特征图聚合得到最终的去雨图像.采用不同分辨率的特征提取网络,能更好地提取雨纹的特征.特征细化网络首先经过多尺度扩张卷积聚合雨纹特征,然后通道注意力给聚合模块通道分配不同的权重以细化自身,最后使用ResNet和DenseNet的双路特征提取器实现雨纹信息提取.仿真和真实雨数据集结果表明:所提网络能够有效地去除雨纹,得到高质量的无雨图像.
|
关 键 词: | 单幅图像去雨 扩张卷积 双路网络 通道注意力 长短时间记忆网络 |
本文献已被 CNKI 万方数据 等数据库收录! |
|