一种基于注意机制的金字塔去雨网络研究

摘要：

下雨作为一种常见的自然现象,在给视觉带来影响的同时会降低计算机视觉算法的性能.为了保证计算机视觉算法在各种环境下的有效性,提出一种基于注意机制的金字塔网络进行单幅图像去雨.利用深度学习网络对图像特征进行充分提取,其中对称跳线保证了特征从低维度向高维度的传递,非局域神经网络保证了图像的长时记忆.利用具有不同尺寸核函数的多通道对图像中的雨条纹进行检测,生成的注意图指导网络对雨条纹区域进行有意识关注.该方法以雨图像为输入,去雨图像为输出,是一种端到端的去雨方法.与现有方法相比,所提出的网络取得了较好的去雨效果.

关键词：深度学习;金字塔结构;注意力机制;图像去雨;图像恢复

所属期刊栏目：35

分类号：TP391(计算技术、计算机技术)

资助基金：国家自然科学基金资助项目;山西省自然科学基金资助项目

在线出版日期：2021-08-10（万方平台首次上网日期，不代表论文的发表时间）

页数：共5页

页码：277-280,287