结合亮度约束的双分支结构暗光图像增强算法  

Dual-Branch Structure Low-Light Image Enhancement Algorithm Combined with Brightness Constraint

作　　者：杨海潮 李新凯 张宏立[1] 孟月 YANG Haichao;LI Xinkai;ZHANG Hongli;MENG Yue(School of Electrical Engineering,Xinjiang University,Urumqi 830017,China)

机构地区：[1]新疆大学电气工程学院,乌鲁木齐830017

出　　处：《计算机工程与应用》2025年第8期250-259,共10页Computer Engineering and Applications

基　　金：新疆维吾尔自治区自然科学基金青年基金(2022D01C86);国家自然科学基金(62263030);北京信息科学与技术国家研究中心开放课题(BR2023KF02011)。

摘　　要：暗光图像增强是图像处理中的一个重要问题,其算法性能会影响人眼视觉感知与后续计算机视觉任务,但现有暗光图像增强算法存在内容信息保留与亮度提升平衡性不好的问题。针对此问题,基于CycleGAN提出一种结合亮度约束的双分支结构暗光图像增强算法。该算法利用Retinex理论思想,将暗光增强任务解耦为内容信息保留与亮度增强两个子任务。构建双分支结构生成器,利用多尺度卷积与稠密连接网络构建内容特征提取分支网络来处理内容信息保留任务;利用空洞卷积与风格校准注意力机制构建亮度均衡分支网络并结合亮度约束来处理亮度增强任务。通过特征融合模块将两个分支网络的输出结果进行融合与解码,得到增强后的图像。此外,通过亮度响应函数来扩充暗光图像数据集,增加训练数据的多样性。实验结果表明:相比于当前主流算法,该算法增强后的暗光图片具有更好的人眼主观视觉效果,同时在PSNR、SSIM、NIQE、Histogram等客观评价指标上也具有明显优势。Low-light image enhancement is an important problem in image processing,and its algorithm performance will affect the human visual perception and subsequent computer vision tasks,but the existing low-light image enhancement algorithms have poor balance between content information retention and brightness enhancement.To solve this problem,a double branch low-light image enhancement algorithm based on CycleGAN is proposed,which combines brightness constraint.Firstly,the algorithm uses Retinex theory to decouple the low-light enhancement task into two sub-tasks:content information retention and brightness enhancement.Secondly,the two-branch structure generator is constructed,and the content feature extraction branch network is constructed by using multi-scale convolution and dense connection network to deal with the content information retention task.The luminance equalization branch network is constructed by using the cavity convolution and style calibration attention mechanism,and the luminance enhancement task is processed by combining the luminance constraint.Finally,the output results of the two branch networks are fused and decoded by the feature fusion module,and the enhanced image is obtained.In addition,a low-light image dataset is expanded with the brightness response function to increase the diversity of training data.The experimental results show that,compared with the current mainstream algorithm,the enhanced low-light images have better subjective visual effects for human eyes,and also have obvious advantages in terms of objective evaluation indicators such as PSNR,SSIM,NIQE,and Histogram.

关 键 词：暗光图像增强 循环生成对抗网络 双分支结构 稠密连接网络 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]