EnlightenGAN: Deep Light Enhancement Without Paired Supervision

被引:1318
作者
Jiang, Yifan [1 ]
Gong, Xinyu [1 ]
Liu, Ding [2 ]
Cheng, Yu [2 ,3 ]
Fang, Chen [2 ]
Shen, Xiaohui [2 ]
Yang, Jianchao [2 ]
Zhou, Pan [4 ]
Wang, Zhangyang [1 ]
机构
[1] Univ Texas Austin, Dept Elect & Comp Engn, Austin, TX 78712 USA
[2] Bytedance Inc, Mountain View, CA 94041 USA
[3] Microsoft AI & Res, Redmond, WA 98052 USA
[4] Huazhong Univ Sci & Technol, Dept Elect Informat & Commun, Wuhan 430074, Peoples R China
关键词
Training; Visualization; Lighting; Generative adversarial networks; Gallium nitride; Adaptation models; Training data; Low-light enhancement; generative adversarial networks; unsupervised learning; RETINEX;
D O I
10.1109/TIP.2021.3051462
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Deep learning-based methods have achieved remarkable success in image restoration and enhancement, but are they still competitive when there is a lack of paired training data? As one such example, this paper explores the low-light image enhancement problem, where in practice it is extremely challenging to simultaneously take a low-light and a normal-light photo of the same visual scene. We propose a highly effective unsupervised generative adversarial network, dubbed EnlightenGAN, that can be trained without low/normal-light image pairs, yet proves to generalize very well on various real-world test images. Instead of supervising the learning using ground truth data, we propose to regularize the unpaired training using the information extracted from the input itself, and benchmark a series of innovations for the low-light image enhancement problem, including a global-local discriminator structure, a self-regularized perceptual loss fusion, and the attention mechanism. Through extensive experiments, our proposed approach outperforms recent methods under a variety of metrics in terms of visual quality and subjective user study. Thanks to the great flexibility brought by unpaired training, EnlightenGAN is demonstrated to be easily adaptable to enhancing real-world images from various domains. Our codes and pre-trained models are available at: https://github.com/VITA-Group/EnlightenGAN.
引用
收藏
页码:2340 / 2349
页数:10
相关论文
共 58 条
[41]   ADAPTIVE HISTOGRAM EQUALIZATION AND ITS VARIATIONS [J].
PIZER, SM ;
AMBURN, EP ;
AUSTIN, JD ;
CROMARTIE, R ;
GESELOWITZ, A ;
GREER, T ;
TERHAARROMENY, B ;
ZIMMERMAN, JB ;
ZUIDERVELD, K .
COMPUTER VISION GRAPHICS AND IMAGE PROCESSING, 1987, 39 (03) :355-368
[42]   Attentive Generative Adversarial Network for Raindrop Removal from A Single Image [J].
Qian, Rui ;
Tan, Robby T. ;
Yang, Wenhan ;
Su, Jiajun ;
Liu, Jiaying .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :2482-2491
[43]   Joint Enhancement and Denoising Method via Sequential Decomposition [J].
Ren, Xutong ;
Li, Mading ;
Cheng, Wen-Huang ;
Liu, Jiaying .
2018 IEEE INTERNATIONAL SYMPOSIUM ON CIRCUITS AND SYSTEMS (ISCAS), 2018,
[44]   PsyPhy: A Psychophysics Driven Evaluation Framework for Visual Recognition [J].
RichardWebster, Brandon ;
Anthony, Samuel E. ;
Scheirer, Walter J. .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2019, 41 (09) :2280-2286
[45]   U-Net: Convolutional Networks for Biomedical Image Segmentation [J].
Ronneberger, Olaf ;
Fischer, Philipp ;
Brox, Thomas .
MEDICAL IMAGE COMPUTING AND COMPUTER-ASSISTED INTERVENTION, PT III, 2015, 9351 :234-241
[46]   Scale-recurrent Network for Deep Image Deblurring [J].
Tao, Xin ;
Gao, Hongyun ;
Shen, Xiaoyong ;
Wang, Jue ;
Jia, Jiaya .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :8174-8182
[47]   Improved Texture Networks: Maximizing Quality and Diversity in Feed-forward Stylization and Texture Synthesis [J].
Ulyanov, Dmitry ;
Vedaldi, Andrea ;
Lempitsky, Victor .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :4105-4113
[48]   Naturalness Preserved Enhancement Algorithm for Non-Uniform Illumination Images [J].
Wang, Shuhang ;
Zheng, Jin ;
Hu, Hai-Miao ;
Li, Bo .
IEEE TRANSACTIONS ON IMAGE PROCESSING, 2013, 22 (09) :3538-3548
[49]  
Wei Cui, 2018, 2018 Photonics North (PN), DOI 10.1109/PN.2018.8438843
[50]   Deep High Dynamic Range Imaging with Large Foreground Motions [J].
Wu, Shangzhe ;
Xu, Jiarui ;
Tai, Yu-Wing ;
Tang, Chi-Keung .
COMPUTER VISION - ECCV 2018, PT II, 2018, 11206 :120-135