多尺度注意力密集连接网络的单幅图像去雨算法 |
| |
引用本文: | 李宏涛,李朋鹏,金桂月,金基宇.多尺度注意力密集连接网络的单幅图像去雨算法[J].大连工业大学学报,2023(6):456-462. |
| |
作者姓名: | 李宏涛 李朋鹏 金桂月 金基宇 |
| |
作者单位: | 大连工业大学信息科学与工程学院 |
| |
摘 要: | 为了获取更好的去雨效果,更好地保留图像纹理细节,设计了一个端到端的多尺度注意力密集连接网络,引入了多尺度注意力模块,通过不同的扩张卷积获得不同尺度下的雨条纹特征。设计了一个并行注意力蒸馏模块,结合通道注意力和子空间注意力机制来校准从多尺度采集获得的图像特征,然后消除冗余特征信息。整体网络架构采用密集连接跨层融合代替直接连接的方式,加强了特征重用,易于训练。结果表明,本方法在提升去雨性能的同时,改善了对于图像的细节保留。
|
关 键 词: | 图像复原 图像去雨 多尺度 注意力机制 知识蒸馏 |
|
|