基于注意力机制及多尺度特征融合的图像去雨

宋建辉, 胡强强, 刘晓阳, 赵亚威

PDF(8885 KB)
PDF(8885 KB)
沈阳理工大学学报 ›› 2024, Vol. 43 ›› Issue (06) : 28-33.

基于注意力机制及多尺度特征融合的图像去雨

  • 宋建辉, 胡强强, 刘晓阳, 赵亚威
作者信息 +
History +

摘要

针对去除不同雨纹的同时恢复图像背景细节的问题,提出一种基于注意力机制及多尺度特征融合的图像去雨方法。该网络采用双分支结构,分别用于雨纹去除和背景恢复。雨纹提取模块采用跨空间学习的多尺度注意力机制,通过多尺度上下文信息捕捉、均值计算、权重计算和整体信息综合,帮助改善雨纹去除任务中的图像质量,提高去雨效果。背景恢复模块包括多尺度特征提取部分和特征融合部分,采用多个扩张卷积层,每个卷积层具有不同的扩张因子,以扩大感受野,提取多尺度的图像背景特征;使用大核卷积对提取的多尺度特征信息进行融合调整,从而更准确地进行背景恢复。在多个公开数据集上的实验结果表明:所提方法能够有效去除真实雨图像场景中的雨纹,同时可以更好地恢复图像背景的细节信息。

关键词

双分支去雨 / 多尺度特征融合 / 注意力机制 / 扩张卷积

中图分类号

TP391.41

引用本文

导出引用
宋建辉, 胡强强, 刘晓阳, 赵亚威. 基于注意力机制及多尺度特征融合的图像去雨. 沈阳理工大学学报. 2024, 43(06): 28-33

基金

辽宁省教育厅高等学校基本科研项目(LJKZ0275); 沈阳市中青年科技创新人才支持计划项目(RC210247)

评论

PDF(8885 KB)

Accesses

Citation

Detail

段落导航
相关文章

/