基于改进注意力机制的低光照图像空间特征门控增强方法

Gate Controlled Enhancement Method for Low-Light Image Spatial Features Based on Improved Attention Mechanism

  • 摘要: 为提升低光照图像在复杂区域的增强效果,解决局部与全局特征利用不充分导致的纹理丢失问题,本文提出空间特征门控复原网络SFGR-Net( spatial feature gated restoration network,SFGR-Net)。首先,纹理强度估计器基于梯度特征动态分割纹理区与平坦区;其次,在特征门控模块中改进非局部注意力机制,建模跨区域长程依赖以提升细节重建精度;进一步设计多尺度残差块,融合新型部分卷积、深度卷积以及普通卷积强化局部特征表示,显著抑制噪声对图像复原干扰。实验结果表明,在LOL、SID和SDSD三个标准数据集上,所提方法PSNR分别达到了23.04dB,23.06dB以及26.92dB,有效保留图像细节信息,展现了较好的增强效果,并且在复杂条件下,模型也具有更高的鲁棒性,可以更好地适用于夜间场景中。

     

    Abstract: To enhance the enhancement effect of low light images in complex areas and solve the problem of texture loss caused by insufficient utilization of local and global features, this paper proposes the Spatial Feature Gated Restoration Network (SFGR Net). Firstly, the texture intensity estimator dynamically segments texture regions and flat regions based on gradient features; Secondly, improve the non local attention mechanism in the feature gating module and model long-range dependencies across regions to enhance the accuracy of detail reconstruction; Further design multi-scale residual blocks, integrating novel partial convolution, deep convolution, and ordinary convolution to enhance local feature representation, significantly suppressing noise interference on image restoration. The experimental results show that the proposed method achieved PSNR of 23.04dB, 23.06dB, and 26.92dB on the three standard datasets of LOL, SID, and SDSD, respectively, effectively preserving image details and demonstrating good enhancement effects. Moreover, under complex conditions, the model also has higher robustness and can be better applied to nighttime scenes.

     

/

返回文章
返回