Laser & Optoelectronics Progress, Volume. 57, Issue 20, 201022(2020)

Low-Light Image Enhancement Based on Attention Mechanism and Convolutional Neural Networks

Ruoyou Wu, Dexing Wang*, and Hongchun Yuan*
Author Affiliations
  • School of Information, Shanghai Ocean University, Shanghai 201306, China
  • show less

    To improve the clarity of low-light images and avoid color distortion, a low-light image enhancement algorithm based on the attention mechanism and convolutional neural network (CNN) is proposed to improve image quality. First, the training data is synthesized based on the Retinex model, and the original image is transformed from RGB (red-green-blue) color space to HSI (hue-saturation-intensity) color space. Then, an A-Unet model is constructed to enhance the brightness component by combining the attention mechanism and CNN. Finally, the enhanced image is obtained by transforming images from the HSI color space to the RGB color space. Experimental results show that the proposed algorithm can effectively improve the image quality, improve the image clarity, and avoid the color distortion. Good results can be obtained in the experiments of synthesizing low-light images and real low-light images, and the subjective and objective evaluation indexes are better than that of the comparison algorithm.

    Tools

    Get Citation

    Copy Citation Text

    Ruoyou Wu, Dexing Wang, Hongchun Yuan. Low-Light Image Enhancement Based on Attention Mechanism and Convolutional Neural Networks[J]. Laser & Optoelectronics Progress, 2020, 57(20): 201022

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Mar. 2, 2020

    Accepted: Apr. 15, 2020

    Published Online: Oct. 14, 2020

    The Author Email: Wang Dexing (dawang@shou.edu.cn), Yuan Hongchun (dawang@shou.edu.cn)

    DOI:10.3788/LOP57.201022

    Topics