基于注意力机制的改进自校准图像增强算法及其在海上低照度场景的应用

Improved self-calibration image enhancement algorithm based on attention mechanism and its application in maritime low-light images

  • 摘要:
    目的 为了解决现有海上低照度图像增强算法存在的亮度提升不足、清晰度低、色彩失真等问题,提出一种基于改进自校准光照学习的海上低照度图像增强算法。
    方法 在自校准光照学习算法的基础上,通过引入注意力机制对低照度图像中光照不均匀区域进行不同程度的增强;构建照明调整模块对光照学习过程的中间输出进行二次开发;引入去噪模块改进黑暗区域的噪声会随着亮度的增强而放大的问题;将批量归一化(BN)改变为批量通道归一化(BCN),该归一化方式利用通道和批次维度,自适应地组合归一化输出。通过主客观两方面进行图像质量评价。
    结果 3个测试集的实验结果表明,改进算法不仅提高了图像亮度,并且增强结果的色彩丰富度较高,无色彩失真;与未改进原始算法相比,标准差平均提升了20.01%,自然图像质量评价降低了9.16%,平均梯度和信息熵分别提升了23.68%和6.46%。
    结论 改进算法在图像视觉质量方面取得了突破,使得不同环境下的海上低照度图像都能达到较好的增强效果。

     

    Abstract:
    Objective In order to solve the problems of the insufficient brightness enhancement, low sharpness, and color distortion of existing maritime low-light image enhancement algorithms, this paper proposes an algorithm based on improved self-calibrated light learning.
    Method An attention mechanism is introduced to enhance the unevenly illuminated regions in low-light images to different degrees; an illumination adjustment module is constructed to secondary the intermediate outputs of the light learning process; a denoising module is introduced to ameliorate the problem of noise in dark regions being amplified with the enhancement of the brightness; and batch normalization (BN) is changed to batch channel normalization (BCN), which utilizes channel and batch dimensions to adaptively combine the normalized outputs. Image quality is then evaluated in both subjective and objective aspects.
    Results Experiments are conducted under three test sets. The results show that when compared with the original unimproved algorithm, the improved algorithm not only improves the image brightness, but also provides higher color richness and no color distortion, while improving the standard deviation by an average of 20.01%, reducing the natural image quality evaluation by 9.16%, and improving the average gradient and information entropy by 23.68% and 6.46% respectively.
    Conclusion The improved algorithm represents a breakthrough in image visual quality, enabling the better enhancement of maritime low-light images in different environments.

     

/

返回文章
返回