[发明专利]适用于单光源的色适应驱动白平衡校正方法、装置、电子设备及存储介质

说明书

本申请属于图像处理技术领域，公开了一种适用于单光源的色适应驱动白平衡校正方法、装置、电子设备及存储介质，能够提高了图像白平衡校正的准确度。该方法具体包括：利用图像采集设备的特性化矩阵和锥体响应变换矩阵，将拍摄图像的色度值和环境光源的色度值，分别转换为图像LMS视锥响应和环境光源LMS视锥响应值；基于所述环境光源LMS视锥响应值及参考光源LMS视锥响应值，计算每一视锥通道的增益系数；利用所述增益系数和图像LMS视锥响应，计算拍摄图像在参考光源下的对应色；利用特性化矩阵和锥体响应变换矩阵的逆矩阵，将所述对应色进行转换，实现拍摄图像的白平衡校正。

技术领域

本发明涉及相机白平衡校正技术领域，具体涉及一种适用于单光源的色适应驱动白平衡校正方法、装置、电子设备及存储介质。

背景技术

人眼视觉系统的调节功能可以部分抵消光源颜色对于物体色貌的影响，在不同的光源下保持物体色貌的相对稳定，这种现象被称为色适应现象。为了尽可能还原人眼的色貌感知，彩色数码相机信号处理流程(ISP)中的白平衡模块承担了色适应调节的功能，根据实际光环境的色度、亮度等信息，对原始图像的颜色进行线性矫正，其流程如图1所示。

相机的白平衡矫正算法包括两个模块：光源估计和色适应调整。其中光源估计模块基于图像颜色信息预估出目前环境中的光源颜色和光环境亮度，然后色适应调整模块根据估算出的实际光源颜色调节图像中每个像素点的响应值。目前常用的光源估计算法包括灰度世界法、最大值缩放法、色域映射法、神经网络法、颜色相关法以及规则模糊法等，所采用的色适应调整算法都基于沃克里斯假设，通过计算实际光源和参考光源颜色在红(R)、绿(G)、蓝(B)三通道的比值得到每个通道的增益系数，并使用该系数对原图像的三通道响应值做独立线性调节，使得在实际场景中拍摄的图像经过白平衡矫正后与参考光源下的图像相匹配。

相机白平衡矫正算法的关键技术之一是开发与人眼视觉系统一致的色适应调整算法。但目前工业界普遍使用的色适应调整算法存在一系列问题：(1)色适应现象是视锥细胞光谱敏感度随光环境变化导致的，但白平衡算法中的图像线性变换并不是在用来描述视锥细胞响应的LMS空间中进行的，而是基于设备相关的RGB空间或者R/G,B/G空间，颜色空间的不匹配会导致颜色还原结果偏离人眼感知；(2)对于任何照明光源和适应背景，现有的白平衡算法假设人眼达到了完全适应的状态，并将参考光源作为矫正目标，而忽视了视觉系统可能存在的不完全适应的情况，也没有考虑光源颜色、亮度、大小等因素对于适应程度的影响，导致白平衡矫正后的图像与人眼视觉之间产生偏差。

发明内容

本申请实施例提供一种适用于单光源的色适应驱动白平衡校正方法、装置、电子设备及存储介质，提高了图像白平衡校正的准确度。

第一方面，本申请一实施例提供了一种适用于单光源的色适应驱动白平衡校正方法，包括：

利用图像采集设备的特性化矩阵和锥体响应变换矩阵，将拍摄图像的色度值和环境光源的色度值，分别转换为图像LMS视锥响应和环境光源LMS视锥响应值；

基于所述环境光源LMS视锥响应值及参考光源LMS视锥响应值，计算每一视锥通道的增益系数；

利用所述增益系数和图像LMS视锥响应，计算拍摄图像在参考光源下的对应色；

利用特性化矩阵和锥体响应变换矩阵的逆矩阵，将所述对应色进行转换，实现拍摄图像的白平衡校正。

可选的，所述基于所述环境光源LMS视锥响应值及参考光源LMS视锥响应值，计算每一视锥通道的增益系数的具体过程为：

基于环境光源LMS视锥响应值，计算环境光源的影响因子；

将所述影响因子与周边环境因子的乘积作为环境光源颜色的适应程度；

基于所述环境光源颜色的适应程度、环境光源LMS视锥响应值及参考光源LMS视锥响应值，计算每一视锥通道的增益系数。