高级检索

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

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 Rratio (PSNR) and structural similarity Iindex (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.

     

/

返回文章
返回