RASP-Net: 基于残差块和高相似传递注意力的图像去雪网络
RASP-Net: Residual Block with High Similarity Pass Attention for Image Desnowing
-
摘要: 在冬季或高海拔的多雪地区, 大量的雪花会覆盖在图像上, 严重影响图像的质量. 当前图像去雪研究中, 部分算法侧重提取海量特征来检测并去除多尺度雪花, 却缺少特征甄别筛选机制. 为了提取相关性强的雪花特征, 本文提出了基于残差块和高相似传递注意力的图像去雪网络RASP-Net, 可清晰展示被雪花遮挡的细节. 首先通过残差块和空间注意力机制的结合, 动态增强样本有用特征抑制无用雪花特征, 同时使用高相似性传递注意力和软阈值处理技术动态地判断雪花特征, 最后通过循环结构细化特征提取和雪花去除, 提升模型的去雪性能和鲁棒性. 大量的实验表明, RASP-Net在合成和真实世界的数据集的去雪效果均优于当前先进的方法.Abstract: In snowy seasons or high-altitude regions with frequent snowfall, images are often heavily obscured by snowflakes, severely degrading their visual quality. While existing image desnowing methods typically focus on extracting abundant features to detect and remove multi-scale snowflakes, they often lack effective mechanisms for feature selection and discrimination. To address this limitation, we propose RASP-Net, a Residual Block and High Similarity Pass Attention based Desnowing Network, which effectively reveals image details obscured by snowflakes. Specifically, RASP-Net integrates residual learning with a spatial attention mechanism to dynamically enhance useful features while suppressing irrelevant snowflake noise. It further incorporates a high-similarity attention mechanism combined with soft-thresholding to adaptively identify snowflake features. Finally, a recurrent refinement module is introduced to progressively enhance feature extraction and snowflake removal, thereby improving both the desnowing performance and the robustness of the model. Extensive experiments demonstrate that RASP-Net consistently outperforms state-of-the-art methods on both synthetic and real-world datasets.