arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2405.17725cs.CV

颜色偏移估计与校正用于图像增强

Color Shift Estimation-and-Correction for Image Enhancement

City University of Hong Kong(香港城市大学)

Yiyu Li, Ke Xu, Gerhard Petrus Hancke, Rynson W. H. Lau

AI总结:

提出一种通过估计和校正颜色偏移来增强同时含过曝与欠曝图像的方法,利用COSE和COMO模块分别处理,实验证明优于现有方法。

摘要:

在次优光照条件下拍摄的图像可能同时包含过度曝光和欠曝光区域。现有方法主要侧重于调整图像亮度,这可能会加剧欠曝光区域的色调失真,并且无法恢复过度曝光区域的准确颜色。我们观察到,过度曝光和欠曝光区域相对于彼此表现出相反的颜色色调分布偏移,由于它们通常没有“正常曝光”区域或像素作为参考,因此这种偏移在联合建模中可能不易被归一化。在本文中,我们提出了一种新颖的方法,通过学习估计和校正此类颜色偏移来增强同时包含过度曝光和欠曝光的图像。具体来说,我们首先通过基于UNet的网络推导出输入图像的亮化和暗化版本的颜色特征图,随后使用伪正常特征生成器生成伪正常颜色特征图。然后,我们提出了一种新颖的颜色偏移估计(COSE)模块,用于估计推导出的亮化(或暗化)颜色特征图与伪正常颜色特征图之间的颜色偏移。COSE模块分别校正过度曝光和欠曝光区域的估计颜色偏移。我们进一步提出了一种新颖的颜色调制(COMO)模块,用于调制过度曝光和欠曝光区域中分别校正的颜色,以生成增强图像。综合实验表明,我们的方法优于现有方法。项目网页:https://github.com/yiyulics/CSEC。

英文摘要:

Images captured under sub-optimal illumination conditions may contain both over- and under-exposures. Current approaches mainly focus on adjusting image brightness, which may exacerbate the color tone distortion in under-exposed areas and fail to restore accurate colors in over-exposed regions. We observe that over- and under-exposed regions display opposite color tone distribution shifts with respect to each other, which may not be easily normalized in joint modeling as they usually do not have ``normal-exposed'' regions/pixels as reference. In this paper, we propose a novel method to enhance images with both over- and under-exposures by learning to estimate and correct such color shifts. Specifically, we first derive the color feature maps of the brightened and darkened versions of the input image via a UNet-based network, followed by a pseudo-normal feature generator to produce pseudo-normal color feature maps. We then propose a novel COlor Shift Estimation (COSE) module to estimate the color shifts between the derived brightened (or darkened) color feature maps and the pseudo-normal color feature maps. The COSE module corrects the estimated color shifts of the over- and under-exposed regions separately. We further propose a novel COlor MOdulation (COMO) module to modulate the separately corrected colors in the over- and under-exposed regions to produce the enhanced image. Comprehensive experiments show that our method outperforms existing approaches. Project webpage: https://github.com/yiyulics/CSEC.

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

↑