通过自适应伽马校正与注意力网络,同时提升暗光图像亮度和细节。
Low-Light Image Enhancement Using Gamma Learning And Attention-Enabled Encoder-Decoder Networks
- 分两阶段:先用自适应伽马校正全局调亮,再用注意力网络修复局部细节。
- 在LOL数据集上达29.96 dB PSNR、0.9458 SSIM,优于现有方法。
- 适合需要高保真暗光图像的场景,如自动驾驶与安防监控。
低光照环境下的图像对计算机视觉系统和人类感知构成重大挑战,尤其在需精确目标识别与场景分析的应用中。这类图像普遍存在噪声放大、光照不足、对比度下降、色彩失真及细节丢失等问题。尽管近期深度学习方法已展现潜力,但如何构建简单高效的框架,自然融合全局光照调整与局部细节增强仍具挑战。为此,我们提出双阶段深度架构:第一阶段采用自适应伽马校正模块(AGCM),基于局部与全局线索为每个像素学习合适伽马值,生成调亮的中间输出;第二阶段使用含卷积块注意力模块(CBAM)的编码器-解码器网络,在此基础上恢复精细细节。训练采用包含L1重构、SSIM、总变差、颜色恒常性及伽马正则化项的复合损失函数,以平衡像素精度与视觉质量。在LOL-v1、LOL-v2真实与合成数据集上的实验表明,本方法达到最高29.96 dB PSNR与0.9458 SSIM,优于现有方法。在DICM、LIME、MEF、NPE数据集上,以NIQE、BRISQUE、UNIQUE指标评估,感知质量更优且伪影更少,所有数据集均取得最佳NIQE分数。所提GAtED(Gamma learned and Attention-enabled Encoder-Decoder)方法有效兼顾全局光照调整与局部细节增强,为暗光增强提供实用解决方案。
原文摘要 · Abstract (English)
Images acquired in low-light environments present significant obstacles for computer vision systems and human perception, especially for applications requiring accurate object recognition and scene analysis. Such images typically manifest multiple quality issues: amplified noise, inadequate scene illumination, contrast reduction, color distortion, and loss of details. While recent deep learning methods have shown promise, developing simple and efficient frameworks that naturally integrate global illumination adjustment with local detail refinement continues to be an important objective. To this end, we introduce a dual-stage deep learning architecture that combines adaptive gamma correction with attention-enhanced refinement to address these fundamental limitations. The first stage uses an Adaptive Gamma Correction Module (AGCM) to learn suitable gamma values for each pixel based on both local and global cues, producing a brightened intermediate output. The second stage applies an encoder-decoder deep network with Convolutional Block Attention Modules (CBAM) to this brightened image, in order to restore finer details. We train the network using a composite loss that includes L1 reconstruction, SSIM, total variation, color constancy, and gamma regularization terms to balance pixel accuracy with visual quality. Experiments on LOL-v1, LOL-v2 real, and LOL-v2 synthetic datasets show our method reaches PSNR of upto 29.96 dB and upto 0.9458 SSIM, outperforming existing approaches. Additional tests on DICM, LIME, MEF, and NPE datasets using NIQE, BRISQUE, and UNIQUE metrics confirm better perceptual quality with fewer artifacts, achieving the best NIQE scores across all datasets. Our GAtED (Gamma learned and Attention-enabled Encoder-Decoder) method effectively handles both global illumination adjustment and local detail enhancement, offering a practical solution for low-light enhancement.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。