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 条
  • [21] Dual Cross-Attention Multi-Stage Embedding Network for Low-Light Image Enhancement
    Fan, Junyu
    Li, Jinjiang
    Hua, Zhen
    JOURNAL OF CIRCUITS SYSTEMS AND COMPUTERS, 2024, 33 (07)
  • [22] Attention-based Broad Self-guided Network for Low-light Image Enhancement
    Chen, Zilong
    Liang, Yaling
    Du, Minghui
    2022 26TH INTERNATIONAL CONFERENCE ON PATTERN RECOGNITION (ICPR), 2022, : 31 - 38
  • [23] Low-light image enhancement algorithm based on multi-channel fusion attention network
    Chen Q.
    Gu Y.
    Guangxue Jingmi Gongcheng/Optics and Precision Engineering, 2023, 31 (14): : 2111 - 2122
  • [24] Wavelet-based enhancement network for low-light image
    Hu, Xiaopeng
    Liu, Kang
    Yin, Xiangchen
    Gao, Xin
    Jiang, Pingsheng
    Qian, Xu
    DISPLAYS, 2025, 87
  • [25] Low-light image enhancement via improved lightweight YUV attention network
    Abbass, Mohammed Y.
    Kasban, H.
    Elsharkawy, Zeinab F.
    COMPUTERS & GRAPHICS-UK, 2025, 127
  • [26] A Joint Network for Low-Light Image Enhancement Based on Retinex
    Jiang, Yonglong
    Zhu, Jiahe
    Li, Liangliang
    Ma, Hongbing
    COGNITIVE COMPUTATION, 2024, 16 (06) : 3241 - 3259
  • [27] Low-Light Image Enhancement Based on Generative Adversarial Network
    Abirami, R. Nandhini
    Vincent, P. M. Durai Raj
    FRONTIERS IN GENETICS, 2021, 12
  • [28] Cyclic Generative Attention-Adversarial Network for Low-Light Image Enhancement
    Zhen, Tong
    Peng, Daxin
    Li, Zhihui
    SENSORS, 2023, 23 (15)
  • [29] Low-Light Image Enhancement Network Using Informative Feature Stretch and Attention
    Chun, Sung Min
    Park, Jun Young
    Eom, Il Kyu
    ELECTRONICS, 2024, 13 (19)
  • [30] 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