基于倍频卷积和注意力机制的图像去雨 |
| |
作者姓名: | 杨青 于明 付强 阎刚 |
| |
作者单位: | 河北工业大学 电子信息工程学院,天津 300401;陆军工程大学石家庄校区 电子与光学工程系,石家庄 050003;河北工业大学 人工智能与数据科学学院,天津 300401 |
| |
基金项目: | 国家自然科学基金项目(61806071,62102129). |
| |
摘 要: | 针对机器视觉场景图像中由于雨线影响导致背景信息模糊、损失的问题,提出一种基于倍频卷积和注意力机制的图像去雨方法.首先,建立基于空-频域去雨模型,设计基于空间尺度变换和倍频卷积的频率特征分解模块,通过学习得到频率特征和雨线特征的映射关系,降低低频特征空间冗余,提高网络运行效率;其次,设计多层通道注意力模块映射雨线层权重信息,增强重要特征,挖掘雨线层之间的亮度差异,提高雨线检测性能;最后,通过序列操作迭代分解出不同成分的雨线信息,进而完成场景图像去雨.实验结果表明,所提方法对不同方向、形状的雨线和雨滴具有良好的去除性能,同时对于背景图像的细节与边缘信息也具有较好的保护作用.
|
关 键 词: | 图像去雨 去雨模型 倍频卷积 分频特征映射 多层通道注意力 特征权重 |
|
| 点击此处可从《控制与决策》浏览原始摘要信息 |
|
点击此处可从《控制与决策》下载全文 |
|