-
题名基于离散注意力机制的信息蒸馏去雨网络研究
- 1
-
-
作者
罗岩
周杰
罗宏
徐蕾
邵根富
-
机构
南京信息工程大学电子与信息工程学院
杭州电子科技大学自动化学院
-
出处
《陕西科技大学学报》
北大核心
2023年第1期180-186,共7页
-
基金
国家自然科学基金项目(61971167,62101275,62101274)。
-
文摘
针对下雨时拍摄的图像会被雨纹等噪声干扰,影响视觉算法的处理结果,提出一种基于离散注意力机制的信息蒸馏去雨网络实现单幅图像去雨.该模型使用空洞卷积构成的池化网络进行数据提取,将各个尺寸的颗粒特征加以集成,在获取最大感受野的同时,使分辨率损失减小.利用循环网络选择性进行特征提取,增加重要特征的权重.通过离散注意力蒸馏模块,逐步细化图像的细节,得到部分细节特征后,将剩下的特征加以进一步的精细处理.算法在数据集Rain100H、Rain100L和自建数据集RS1000中进行测试,峰值信噪比和结构相似性分别达到30.06/0.913、38.61/0.983、33.86/0.950.实验数据表明,该算法在两种指标上皆优于对比算法,细节信息保留的更加完全.
-
关键词
图像去雨
信息蒸馏
多尺度
离散注意力机制
-
Keywords
image rain removal
information distillation
multi-scale
discrete attention mechanism
-
分类号
TP391.41
[自动化与计算机技术—计算机应用技术]
-
