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 条
  • [31] Lightening Network for Low-Light Image Enhancement
    Wang, Li-Wen
    Liu, Zhi-Song
    Siu, Wan-Chi
    Lun, Daniel P. K.
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2020, 29 : 7984 - 7996
  • [32] Flow Learning Based Dual Networks for Low-Light Image Enhancement
    Wang, Siyu
    Hu, Changhui
    Yi, Weilin
    Cai, Ziyun
    Zhai, Mingliang
    Yang, Wankou
    NEURAL PROCESSING LETTERS, 2023, 55 (06) : 8115 - 8130
  • [33] Adaptive Dual Aggregation Network with Normalizing Flows for Low-Light Image Enhancement
    Wang, Hua
    Cao, Jianzhong
    Huang, Jijiang
    ENTROPY, 2024, 26 (03)
  • [34] An illumination-guided dual attention vision transformer for low-light image enhancement
    Wen, Yanjie
    Xu, Ping
    Li, Zhihong
    Xu, Wangtu
    PATTERN RECOGNITION, 2025, 158
  • [35] Flow Learning Based Dual Networks for Low-Light Image Enhancement
    Siyu Wang
    Changhui Hu
    Weilin Yi
    Ziyun Cai
    Mingliang Zhai
    Wankou Yang
    Neural Processing Letters, 2023, 55 : 8115 - 8130
  • [36] Dual-band low-light image enhancement
    Aizhong Mi
    Wenhui Luo
    Zhanqiang Huo
    Multimedia Systems, 2024, 30
  • [37] Two-stage low-light image enhancement network with an attention mechanism and cross-stage connection
    Kuang, Boan
    Zhang, Zhibin
    JOURNAL OF ELECTRONIC IMAGING, 2022, 31 (05)
  • [38] Dual-band low-light image enhancement
    Mi, Aizhong
    Luo, Wenhui
    Huo, Zhanqiang
    MULTIMEDIA SYSTEMS, 2024, 30 (02)
  • [39] WMANet: Wavelet-Based Multi-Scale Attention Network for Low-Light Image Enhancement
    Xiang, Yangjun
    Hu, Gengsheng
    Chen, Mei
    Emam, Mahmoud
    IEEE ACCESS, 2024, 12 : 105674 - 105685
  • [40] Attention Guided Low-Light Image Enhancement with a Large Scale Low-Light Simulation Dataset
    Lv, Feifan
    Li, Yu
    Lu, Feng
    INTERNATIONAL JOURNAL OF COMPUTER VISION, 2021, 129 (07) : 2175 - 2193