Transformer-Based Multi-scale Optimization Network for Low-Light Image Enhancement

被引:0
|
作者
Niu Y. [1 ]
Lin X. [1 ]
Xu H. [1 ]
Li Y. [1 ]
Chen Y. [1 ]
机构
[1] College of Computer and Data Science, Fuzhou University, Fuzhou
来源
Moshi Shibie yu Rengong Zhineng/Pattern Recognition and Artificial Intelligence | 2023年 / 36卷 / 06期
基金
中国国家自然科学基金;
关键词
Attention Mechanism; Image Enhancement; Low-Light Image Enhancement; Multi-scale Feature Fusion;
D O I
10.16451/j.cnki.issn1003-6059.202306003
中图分类号
学科分类号
摘要
Enhancing low-light images with high quality is a highly challenging task due to the features of low-light images such as brightness, color, and details in the information of different scales. Existing deep learning-based methods fail to fully utilize multi-scale features and fuse multi-scale features to comprehensively enhance the brightness, color and details of the images. To address these problems, a Transformer-based multi-scale optimization network for low-light image enhancement is proposed. Firstly, the Transformer-based multi-task enhancement module is designed. Through multi-task training, the Transformer-based enhancement module gains the ability to globally model brightness, color, and details. Therefore, it can initially cope with various degradation challenges commonly found in low-light images, such as insufficient brightness, color deviation, blurred details and severe noises. Then, the architecture combining global and local multi-scale features is designed to progressively optimize the features at different scales. Finally, a multi-scale feature fusion module and an adaptive enhancement module are proposed. They learn and fuse the information association among different scales, while adaptively enhancing images in various local multi-scale branches. Extensive experiments on six public datasets, including paired or unpaired images, show that the proposed method can effectively solve the problems of multiple degradation types, such as brightness, color, details and noise in low-light images. © 2023 Journal of Pattern Recognition and Artificial Intelligence. All rights reserved.
引用
收藏
页码:511 / 529
页数:18
相关论文
共 57 条
  • [1] LI C Y, GUO C L, HAN L H, Et al., Low-Light Image and Video Enhancement Using Deep Learning: A Survey, IEEE Transactions on Pattern Analysis and Machine Intelligence, 44, 12, pp. 9396-9416, (2021)
  • [2] LU F F, LI Y, LU F., Attention Guided Low-Light Image Enhancement with a Large Scale Low-Light Simulation Dataset, International Journal of Computer Vision, 129, 7, pp. 2175-2193, (2021)
  • [3] ARICI T, DIKBAS S, ALTUNBASAK Y., A Histogram Modification Framework and Its Application for Image Contrast Enhancement, IEEE Transactions on Image Processing, 18, 9, pp. 1921-1935, (2009)
  • [4] CELIK T, TJAHJADI T., Contextual and Variational Contrast Enhancement, IEEE Transactions on Image Processing, 20, 12, pp. 3431-3441, (2011)
  • [5] XU H T, ZHAI G T, WU X L, Et al., Generalized Equalization Model for Image Enhancement, IEEE Transactions on Multimedia, 16, 1, pp. 68-82, (2013)
  • [6] LEE C, LEE C, KIM C S., Contrast Enhancement Based on Layered Difference Representation of 2D Histograms, IEEE Transactions on Image Processing, 22, 12, pp. 5372-5384, (2013)
  • [7] FU X Y, ZENG D L, HUANG Y, Et al., A Fusion-Based Enhancing Method for Weakly Illuminated Images, Signal Processing, 129, pp. 82-96, (2016)
  • [8] FU X Y, ZENG D L, HUANG Y, Et al., A Weighted Variational Model for Simultaneous Reflectance and Illumination Estimation, Proc of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 2782-2790, (2016)
  • [9] GUO X J, LI Y, LING H B., LIME: Low-Light Image Enhancement via Illumination Map Estimation, IEEE Transactions on Image Processing, 26, 2, pp. 982-993, (2016)
  • [10] LI M D, LIU J Y, YANG W H, Et al., Structure-Revealing Low-Light Image Enhancement via Robust Retinex Model, IEEE Transactions on Image Processing, 27, 6, pp. 2828-2841, (2018)