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

2502025，61（8）ComputerEngineeringandApplications计算机工程与应用

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

+

杨海潮，李新凯，张宏立，孟月

新疆大学电气工程学院，乌鲁木齐830017

+通信作者E-mail：lxk@xju.edu.cn

摘要：暗光图像增强是图像处理中的一个重要问题，其算法性能会影响人眼视觉感知与后续计算机视觉任务，但

现有暗光图像增强算法存在内容信息保留与亮度提升平衡性不好的问题。针对此问题，基于CycleGAN提出一种结

合亮度约束的双分支结构暗光图像增强算法。该算法利用Retinex理论思想，将暗光增强任务解耦为内容信息保留

与亮度增强两个子任务。构建双分支结构生成器，利用多尺度卷积与稠密连接网络构建内容特征提取分支网络来

处理内容信息保留任务；利用空洞卷积与风格校准注意力机制构建亮度均衡分支网络并结合亮度约束来处理亮度

增强任务。通过特征融合模块将两个分支网络的输出结果进行融合与解码，得到增强后的图像。此外，通过亮度响

应函数来扩充暗光图像数据集，增加训练数据的多样性。实验结果表明：相比于当前主流算法，该算法增强后的暗

光图片具有更好的人眼主观视觉效果，同时在PSNR、SSIM、NIQE、Histogram等客观评价指标上也具有明显优势。

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

文献标志码：A中图分类号：TP391.4doi：10.3778/j.issn.1002-8331.2312-0264

Dual-BranchStructureLow-LightImageEnhancementAlgorithmCombinedwithBrightness

Constraint

+

YANGHaichao,LIXinkai,ZHANGHongli,MENGYue

SchoolofElectricalEngineering,XinjiangUniversity,Urumqi830017,China

Abstract：Low-lightimageenhancementisanimportantprobleminimageprocessing,anditsalgorithmperformance

willaffectthehumanvisualperceptionandsubsequentcomputervisiontasks,buttheexistinglow-lightimageenhance-

mentalgorithmshavepoorbalancebetweencontentinformationretentionandbrightnessenhancement.Tosolvethisprob-

lem,adoublebranchlow-lightimageenhancementalgorithmbasedonCycleGANisproposed,whichcombinesbright-

nessconstraint.Firstly,thealgorithmusesRetinextheorytodecouplethelow-lightenhancementtaskintotwosub-tasks:

contentinformationretentionandbrightnessenhancement.Secondly,thetwo-branchstructuregeneratorisconstructed,

andthecontentfeatureextractionbranchnetworkisconstructedbyusingmulti-scaleconvolutionanddenseconnection

networktodealwiththecontentinformationretentiontask.Theluminanceequalizationbranchnetworkisconstructedby

usingthecavitycon