计算机科学与探索 ›› 2024, Vol. 18 ›› Issue (9): 2422-2435.DOI: 10.3778/j.issn.1673-9418.2310007

• 图形·图像 • 上一篇    下一篇

LEGAN:一种新的暗弱光照图像增强算法

郭璠,刘文韬,李小虎,唐琎   

  1. 中南大学 自动化学院,长沙 410083
  • 出版日期:2024-09-01 发布日期:2024-09-01

LEGAN: New Dark and Weak Light Image Enhancement Algorithm

GUO Fan, LIU Wentao, LI Xiaohu, TANG Jin   

  1. School of Automation, Central South University, Changsha 410083, China
  • Online:2024-09-01 Published:2024-09-01

摘要: 针对暗弱光照图像所存在的亮度、对比度、信噪比低,以及噪声污染大等问题,提出了一种新的暗弱光照图像增强算法LEGAN。该算法将图像输入至所提伽马曲线估计网络求得包含伽马参数的特征图,再经过LEB模块增强亮度,并通过级联LEB的方式迭代增强结果。采用基于PatchGAN的全局-局部判别器结构来提高图像分辨率和恢复图像细节。通过引入感知损失来限制真实标签和输出结果之间的差距,利用照明平滑度损失保持相邻像素之间的单调性关系,同时结合空间一致性损失来增强图像的空间相关性。实验结果表明,相比于现今大多数主流增强算法,该算法的细节还原度相对较高,且能有效避免增强后的图像出现局部亮度不佳等问题。

关键词: 暗弱光照, 图像增强, 伽马曲线估计网络, 全局-局部判别器, 损失函数

Abstract: In order to solve the problems of low brightness, contrast, signal-to-noise ratio and high noise pollution in dark and weak light images, this paper proposes a new dark and weak light image enhancement algorithm LEGAN (low-light enhancement generative adversarial network). This algorithm first inputs the image into the proposed Gamma curve estimation network to obtain the feature map containing Gamma parameters, then enhances the brightness through LEB (light enhancement block) module, and iteratively enhances the result by cascading LEB. Then, a global-local discriminator structure based on PatchGAN is used to improve image resolution and recover details. Finally, the gap between the true label and the output result is limited by introducing a perceived loss. The lighting smoothness loss is used to maintain the monotonic relationship between adjacent pixels. Meanwhile, the spatial consistency loss is combined to enhance the spatial correlation of images. Experimental results show that compared with most of the current mainstream enhancement algorithms, the proposed algorithm has a higher degree of detail restoration, and can effectively avoid the problem of ill illumination in the local region of enhanced image.

Key words: dark and weak light, image enhancement, Gamma curve estimation network, global-local discriminator, loss function