高级检索

融合多尺度残差与注意力机制的图像去雾算法

Image Dehazing Algorithm Integrating Multiscale Residual and Attention Mechanism

  • 摘要: 针对现有深度学习去雾算法因编码器与解码器间跨层特征失配,导致细节丢失与对比度下降的问题,提出一种融合多尺度残差与注意力机制的图像去雾算法。首先,设计多尺度残差感知下采样模块,通过多分支并行卷积融合残差连接与高效通道注意力机制,以增强细节特征提取能力;其次,引入自适应细粒度通道注意力机制,动态调整全局与局部信息的特征权重,从而有效抑制对比度下降。在RESIDE−6K、SOTS和HSTS数据集上的实验结果表明:本文算法在主观视觉质量及峰值信噪比(PSNR)、结构相似性指数(SSIM)等客观指标上均优于6种主流对比算法,展现出较强的泛化能力。其中,在RESIDE−6K测试集上,PSNR与SSIM分别达到28.51 dB,0.965 1;在SOTS测试集上分别达到27.82 dB,0.965 5;在HSTS测试集上分别达到29.30 dB,0.960 1。本研究通过多尺度残差与注意力机制的有效协同,实现了细节保留与对比度恢复的双重优化,为有雾图像质量增强提供了有效的技术方案。

     

    Abstract: To address the loss of detail and reduced contrast in foggy images caused by cross-layer feature mismatch between encoders and decoders in existing deep learning algorithms, an image defogging algorithm that integrates multi-scale residuals with attention mechanisms was proposed. Firstly, a multi-scale residual-aware subsampling module was designed at the encoder side. which enhanced detailed feature extraction capability through multi-branch parallel convolutions combined with residual connections and efficient channel attention mechanisms. Secondly, an adaptive fine-grained channel attention mechanism was introduced to dynamically adjust feature weights between global and local information, thereby suppressing contrast degradation. Experimental results on the RESIDE–6K, SOTS, and HSTS datasets demonstrate that the proposed algorithm outperforms six mainstream comparative algorithms in both subjective visual effects and objective metrics such as peak signal-to-noise ratio (PSNR) and structural similarity index (SSIM), while exhibiting strong generalization capability. Specifically, PSNR and SSIM values of 28.51 dB and 0.965 1 are achieved on the RESIDE−6K test set, 27.82 dB and 0.965 5 on the SOTS test set, and 29.30 dB and 0.960 1 on the HSTS test set. Through the effective integration of multi-scale residual and attention mechanisms, dual optimization of detail preservation and contrast recovery is realized, providing an effective technical solution for enhancing the quality of hazy images.

     

/

返回文章
返回