基于动态金字塔和注意力机制的红外图像去噪

Infrared Image Denoising Based on Dynamic Pyramid and Attention Mechanism

  • 摘要: 目前红外图像广泛应用于各个领域,然而由于传感器的本底噪声,红外图像信噪比较低、视觉效果较差。针对现有的图像去噪算法中存在的噪声去除不完全、计算量过大以及纹理特征丢失的问题,本文提出了一种基于动态金字塔和注意力机制的红外图像去噪算法。首先,通过金字塔结构提取图像在不同尺度上的特征;其次,提出了一种动态校准融合机制,提升网络在融合多尺度特征方面的能力;最后,构建了一种局部-上下文注意力块提升网络对局部和上下文信息的恢复能力。在可见光图像数据集和红外图像数据集上的测试结果表明,该算法能够有效去除噪声,保留纹理细节,避免产生伪影斑点,并且相较于NAFNet方法,GFLOPS可降低65%。

     

    Abstract: Infrared images are widely utilized across various applications; however, sensor-induced background noise often results in a low signal-to-noise ratio and poor visual quality. To address issues such as incomplete noise removal, high computational cost, and loss of texture features in existing denoising algorithms, this study proposes an infrared image denoising method based on a dynamic pyramid structure and attention mechanisms. First, multi-scale image features are extracted using a pyramid architecture. Second, a dynamic correction and fusion mechanism is introduced to enhance the network's capability for multi-scale feature integration. Finally, a local-context attention block is designed to enhance the restoration of both local details and contextual information. Experimental results on both visible-light and infrared image datasets demonstrate that the proposed algorithm effectively removes noise, preserves texture details, avoids artifacts and speckle noise, and reduces GFLOPS by 65% compared with the NAFNet method.

     

/

返回文章
返回