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

来源：计算机工程与应用杂志2025年第8期北京时间：

作者：杨海潮, 李新凯, 张宏立, 孟月

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

摘要：暗光图像增强是图像处理中的一个重要问题，其算法性能会影响人眼视觉感知与后续计算机视觉任务，但现有暗光图像增强算法存在内容信息保留与亮度提升平衡性不好的问题。针对此问题，基于CycleGAN提出一种结合亮度约束的双分支结构暗光图像增强算法。该算法利用Retinex理论思想，将暗光增强任务解耦为内容信息保留与亮度增强两个子任务。构建双分支结构生成器，利用多尺度卷积与稠密连接网络构建内容特征提取分支网络来处理内容信息保留任务；利用空洞卷积与风格校准注意力机制构建亮度均衡分支网络并结合亮度约束来处理亮度增强任务。通过特征融合模块将两个分支网络的输出结果进行融合与解码，得到增强后的图像。此外，通过亮度响应函数来扩充暗光图像数据集，增加训练数据的多样性。实验结果表明：相比于当前主流算法，该算法增强后的暗光图片具有更好的人眼主观视觉效果，同时在PSNR、SSIM、NIQE、Histogram等客观评价指标上也具有明显优势。

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