arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.04586cs.CV
收藏

PIE:受物理学启发的低光照增强

PIE: Physics-inspired Low-light Enhancement

College of Computer Science and Technology, Nanjing University of Aeronautics and Astronautics(南京航空航天大学计算机科学与技术学院)

Dong Liang, Zhengyan Xu, Ling Li, Mingqiang Wei, Songcan Chen

AI总结:

本文提出受物理成像原理启发的对比学习低光照增强方法PIE,利用非配对数据、BoC负样本生成和无监督区域分割,在六个跨场景数据集上超过现有最先进模型。

摘要:

本文提出了一种受物理学启发的对比学习范式用于低光照增强,称为PIE。PIE主要解决三个问题:(i)为解决现有基于学习的方法常使用严格像素对应图像对训练LLE模型的问题,我们消除了对像素对应配对训练数据的需求,转而使用非配对图像进行训练。(ii)针对现有方法忽视负样本且负样本生成不充分的问题,我们将受物理学启发的对比学习引入LLE,并设计了Bag of Curves(BoC)方法,以生成更合理、紧密遵循底层物理成像原理的负样本。(iii)为克服现有方法对语义真值的依赖,我们提出了一个无监督区域分割模块,在确保区域亮度一致性的同时消除对语义真值的依赖。总体而言,所提出的PIE能够有效地从非配对正/负样本中学习,并顺利实现非语义区域增强,这与现有LLE工作有明显区别。除了PIE的新颖架构外,我们还探究了PIE在语义分割和人脸检测等下游任务上的增益。在易于获取的开放数据上进行的训练以及大量实验表明,我们的方法在六个独立跨场景数据集上超越了最先进的LLE模型。PIE在测试时运行速度快且GFLOPs合理,便于在移动设备上使用。

英文摘要:

In this paper, we propose a physics-inspired contrastive learning paradigm for low-light enhancement, called PIE. PIE primarily addresses three issues: (i) To resolve the problem of existing learning-based methods often training a LLE model with strict pixel-correspondence image pairs, we eliminate the need for pixel-correspondence paired training data and instead train with unpaired images. (ii) To address the disregard for negative samples and the inadequacy of their generation in existing methods, we incorporate physics-inspired contrastive learning for LLE and design the Bag of Curves (BoC) method to generate more reasonable negative samples that closely adhere to the underlying physical imaging principle. (iii) To overcome the reliance on semantic ground truths in existing methods, we propose an unsupervised regional segmentation module, ensuring regional brightness consistency while eliminating the dependency on semantic ground truths. Overall, the proposed PIE can effectively learn from unpaired positive/negative samples and smoothly realize non-semantic regional enhancement, which is clearly different from existing LLE efforts. Besides the novel architecture of PIE, we explore the gain of PIE on downstream tasks such as semantic segmentation and face detection. Training on readily available open data and extensive experiments demonstrate that our method surpasses the state-of-the-art LLE models over six independent cross-scenes datasets. PIE runs fast with reasonable GFLOPs in test time, making it easy to use on mobile devices.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑