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

被引:4
|
作者
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 条
  • [1] Dual UNet low-light image enhancement network based on attention mechanism
    Fangjin Liu
    Zhen Hua
    Jinjiang Li
    Linwei Fan
    [J]. Multimedia Tools and Applications, 2023, 82 : 24707 - 24742
  • [2] Retinex low-light image enhancement network based on attention mechanism
    Chen, Xinyu
    Li, Jinjiang
    Hua, Zhen
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (03) : 4235 - 4255
  • [3] Retinex low-light image enhancement network based on attention mechanism
    Xinyu Chen
    Jinjiang Li
    Zhen Hua
    [J]. Multimedia Tools and Applications, 2023, 82 : 4235 - 4255
  • [4] ATTENTION-BASED NETWORK FOR LOW-LIGHT IMAGE ENHANCEMENT
    Zhang, Cheng
    Yan, Qingsen
    Zhu, Yu
    Li, Xianjun
    Sun, Jinqiu
    Zhang, Yanning
    [J]. 2020 IEEE INTERNATIONAL CONFERENCE ON MULTIMEDIA AND EXPO (ICME), 2020,
  • [5] Low-Light Image Enhancement using Retinex-based Network with Attention Mechanism
    Ma, Shaojin
    Pan, Weiguo
    Li, Nuoya
    Du, Songjie
    Liu, Hongzhe
    Xu, Bingxin
    Xu, Cheng
    Li, Xuewei
    [J]. International Journal of Advanced Computer Science and Applications, 2024, 15 (01) : 489 - 497
  • [6] LL-UNet++:UNet++ Based Nested Skip Connections Network for Low-Light Image Enhancement
    Shi, Pengfei
    Xu, Xiwang
    Fan, Xinnan
    Yang, Xudong
    Xin, Yuanxue
    [J]. IEEE TRANSACTIONS ON COMPUTATIONAL IMAGING, 2024, 10 : 510 - 521
  • [7] Attention-based dual-color space fusion network for low-light image enhancement
    Huang, Zhixiong
    Li, Jinjiang
    Hua, Zhen
    Fan, Linwei
    [J]. SIGNAL PROCESSING-IMAGE COMMUNICATION, 2023, 119
  • [8] Low-light image enhancement based on GAN with attention mechanism and color Constancy
    Wang, Xiaohong
    Zhai, Yanxiu
    Ma, Xiangcai
    Zeng, Jing
    Liang, Youci
    [J]. MULTIMEDIA TOOLS AND APPLICATIONS, 2022, 83 (1) : 3133 - 3151
  • [9] Low-light image enhancement based on GAN with attention mechanism and color Constancy
    Wang, Xiaohong
    Zhai, Yanxiu
    Ma, Xiangcai
    Zeng, Jing
    Liang, Youci
    [J]. Multimedia Tools and Applications, 2024, 83 (01) : 3133 - 3151
  • [10] Low-light image enhancement based on GAN with attention mechanism and color Constancy
    Xiaohong Wang
    Yanxiu Zhai
    Xiangcai Ma
    Jing Zeng
    Youci Liang
    [J]. Multimedia Tools and Applications, 2024, 83 : 3133 - 3151