Swin transformer and ResNet based deep networks for low-light image enhancement

被引:7
作者
Xu, Lintao [1 ,2 ]
Hu, Changhui [1 ,2 ]
Zhang, Bo [1 ,2 ]
Wu, Fei [1 ,2 ]
Cai, Ziyun [1 ,2 ]
机构
[1] Nanjing Univ Posts & Telecommun, Coll Automat, Wenyuan Rd, Nanjing 210023, Jiangsu, Peoples R China
[2] Nanjing Univ Posts & Telecommun, Coll Artificial Intelligence, Wenyuan Rd, Nanjing 210023, Jiangsu, Peoples R China
基金
中国国家自然科学基金;
关键词
Low-light image enhancement; Generative adversarial network; Swin transformer; Random paired learning; QUALITY ASSESSMENT; RETINEX;
D O I
10.1007/s11042-023-16650-w
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Low-light image enhancement is a long-term low-level vision problem, which aims to improve the visual quality of images captured in low illumination environment. Convolutional neural network (CNN) is the foundation of the majority of low-light image enhancement algorithms now. The limitations of CNN receptive field lead to the inability to establish long-range context interaction. In recent years, Transformer has received increasing attention in computer vision due to its global attention. In this paper, we design the Swin Transformer and ResNet-based Generative Adversarial Network (STRN) for low-light image enhancement by combining the advantages of ResNet and the Swin Transformer. The STRN consists of a U-shaped generator and multiscale discriminators. The generator is composed of a shallow feature extraction, a deep feature extraction, and an image reconstruction module. To calculate the global and local attention, we alternately use Swin Transformer blocks and ResNet in the deep feature processing module. The self perceptual loss and the spatial consistency loss are employed to constrain the random paired training of STRN. The experimental results on benchmark datasets and real-world low-light images demonstrate that the proposed STRN achieves state-of-the-art performance on low-light image enhancement tasks in terms of visual quality and evaluation metrics.
引用
收藏
页码:26621 / 26642
页数:22
相关论文
共 57 条
[1]   A dynamic histogram equalization for image contrast enhancement [J].
Abdullah-Al-Wadud, M. ;
Kabir, Md. Hasanul ;
Dewan, M. Ali Akber ;
Chae, Oksam .
IEEE TRANSACTIONS ON CONSUMER ELECTRONICS, 2007, 53 (02) :593-600
[2]  
Agilandeeswari L, 2022, MULTIMED TOOLS APPL, P1
[3]  
Bychkovsky V, 2011, PROC CVPR IEEE, P97
[4]   Retinex low-light image enhancement network based on attention mechanism [J].
Chen, Xinyu ;
Li, Jinjiang ;
Hua, Zhen .
MULTIMEDIA TOOLS AND APPLICATIONS, 2023, 82 (03) :4235-4255
[5]   Image denoising by sparse 3-D transform-domain collaborative filtering [J].
Dabov, Kostadin ;
Foi, Alessandro ;
Katkovnik, Vladimir ;
Egiazarian, Karen .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2007, 16 (08) :2080-2095
[6]  
Dosovitskiy A., 2020, INT C LEARN REPR, P1
[7]   A weighted variational model for simultaneous reflectance and illumination estimation [J].
Fu, Xueyang ;
Zeng, Delu ;
Huang, Yue ;
Zhang, Xiao-Ping ;
Ding, Xinghao .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :2782-2790
[8]   Deep Bilateral Learning for Real-Time Image Enhancement [J].
Gharbi, Michael ;
Chen, Jiawen ;
Barron, Jonathan T. ;
Hasinoff, Samuel W. ;
Durand, Fredo .
ACM TRANSACTIONS ON GRAPHICS, 2017, 36 (04)
[9]   Zero-Reference Deep Curve Estimation for Low-Light Image Enhancement [J].
Guo, Chunle ;
Li, Chongyi ;
Guo, Jichang ;
Loy, Chen Change ;
Hou, Junhui ;
Kwong, Sam ;
Cong, Runmin .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1777-1786
[10]   Low-light Image Enhancement via Breaking Down the Darkness [J].
Guo, Xiaojie ;
Hu, Qiming .
INTERNATIONAL JOURNAL OF COMPUTER VISION, 2023, 131 (01) :48-66