Dual UNet low-light image enhancement network based on attention mechanism

被引:8
|
作者
Liu, Fangjin [1 ]
Hua, Zhen [1 ]
Li, Jinjiang [2 ]
Fan, Linwei [3 ]
机构
[1] Shandong Technol & Business Univ, Sch Informat & Elect Engn, Yantai 264005, Peoples R China
[2] Shandong Technol & Business Univ, Sch Comp Sci & Technol, Yantai 264005, Peoples R China
[3] Shandong Univ Finance & Econ, Sch Comp Sci & Technol, Jinan, Shandong, Peoples R China
基金
中国国家自然科学基金;
关键词
Low light enhancement; LBP algorithm; Double UNet; Attention model; Recursive calculation; QUALITY ASSESSMENT; SCALE;
D O I
10.1007/s11042-022-14210-2
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Low-light image enhancement has been an important research direction in the field of image processing. Recently, U-Net networks have shown better promise in low-light image enhancement. However, because of the semantic gap and the lack of connection between global contextual information in the U-shaped network, it leads to problems such as inaccurate color information in the enhanced images. To address the above problems, this paper proposes a Dual UNet low-light image enhancement network (DUAMNet) based on an attention mechanism. Firstly, the local texture features of the original image are extracted using the Local Binary Pattern(LBP) operator, and the illumination invariance of the LBP operator better maintains the texture information of the original image. Next, use the Brightness Enhancement Module(BEM). In the BEM module, the outer U-Net network captures feature information at different levels and luminance information of different regions, and the inner densely connected U-Net++ network enhances the correlation of feature information at different levels, mines more hidden feature information extracted by the encoder, and reduces the feature semantic gap between the encoder and decoder. The attention module Convolutional Block Attention Module(CBAM) is introduced in the decoder of U-Net++ network. CBAM further enhances the ability to model the global contextual information linkage and effectively improves the network's attention to the weak light region. The network adopts a progressive recursive structure. The entire network includes four recursive units, and the output of the previous recursive unit is used as the input of the next recursive unit. Comparative experiments are conducted on seven public datasets, and the results are analyzed quantitatively and qualitatively. The results show that despite the simple structure of the network in this paper, the network in this paper outperforms other methods in image quality compared to other methods.
引用
收藏
页码:24707 / 24742
页数:36
相关论文
共 50 条
  • [11] Low-Light Image Enhancement Based on Attention Mechanism and Convolutional Neural Networks
    Wu Ruoyou
    Wang Dexing
    Yuan Hongchun
    LASER & OPTOELECTRONICS PROGRESS, 2020, 57 (20)
  • [12] Channel splitting attention network for low-light image enhancement
    Lu, Bibo
    Pang, Zebang
    Gu, Yanan
    Zheng, Yanmei
    IET IMAGE PROCESSING, 2022, 16 (05) : 1403 - 1414
  • [13] Channel Self-Attention Based Low-Light Image Enhancement Network *
    Wang, Yan
    Su, Peng
    Pan, Xiaoying
    Wang, Hongyu
    Gao, Yuan
    COMPUTERS & GRAPHICS-UK, 2024, 120
  • [14] Dual Autoencoder Network for Retinex-Based Low-Light Image Enhancement
    Park, Seonhee
    Yu, Soohwan
    Kim, Minseo
    Park, Kwanwoo
    Paik, Joonki
    IEEE ACCESS, 2018, 6 : 22084 - 22093
  • [15] Combining Visual Saliency and Attention Mechanism for Low-Light Image Enhancement
    Shang X.
    An N.
    Shang J.
    Zhang S.
    Ding N.
    Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence, 2022, 35 (07): : 602 - 613
  • [16] Low-Light Image Enhancement Network Based on Recursive Network
    Liu, Fangjin
    Hua, Zhen
    Li, Jinjiang
    Fan, Linwei
    FRONTIERS IN NEUROROBOTICS, 2022, 16
  • [17] Color-wise Attention Network for Low-light Image Enhancement
    Atoum, Yousef
    Ye, Mao
    Ren, Liu
    Tai, Ying
    Liu, Xiaoming
    2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION WORKSHOPS (CVPRW 2020), 2020, : 2130 - 2139
  • [18] Underwater low-light enhancement network based on bright channel prior and attention mechanism
    Zheng, Zhangjing
    Huang, Xixia
    Wang, Le
    PLOS ONE, 2023, 18 (02):
  • [19] LERANet: Low-light Enhancement Network based on Retinex and Attention
    He, Renjie
    Guo, Xintao
    Zhou, Wei
    He, Mingyi
    PROCEEDINGS OF THE 2021 IEEE 16TH CONFERENCE ON INDUSTRIAL ELECTRONICS AND APPLICATIONS (ICIEA 2021), 2021, : 1444 - 1449
  • [20] Progressive Dual-Branch Network for Low-Light Image Enhancement
    Cui, Hengshuai
    Li, Jinjiang
    Hua, Zhen
    Fan, Linwei
    IEEE TRANSACTIONS ON INSTRUMENTATION AND MEASUREMENT, 2022, 71