[发明专利]图像的白平衡处理方法、装置、电子设备和存储介质

权利要求书

1.一种图像的白平衡处理方法，其特征在于，所述方法包括：

获取第一色彩空间下的待处理图像；

将所述待处理图像的色彩空间转换为第二色彩空间，得到转换后图像；

基于所述待处理图像和所述转换后图像，确定影响限制信息，其中，所述影响限制信息用以限制色彩空间转换对白平衡处理的影响，所述影响限制信息基于对所述待处理图像进行光源估计得到的第一光源信息和对所述转换后图像进行光源估计得到的第二光源信息确定，或由预训练的影响限制信息确定模型基于所述待处理图像和所述转换后图像确定；

基于所述影响限制信息，对所述待处理图像进行白平衡处理，生成白平衡处理后图像，包括：将所述待处理图像依次右乘所述影响限制信息、第一转换矩阵和第三转换矩阵，得到所述白平衡处理后图像，其中，所述第一转换矩阵用于将所述待处理图像转换为所述待处理图像的白平衡处理后图像，所述第三转换矩阵用于将所述第一色彩空间下的图像转换为所述第二色彩空间下的图像。

2.根据权利要求1所述的方法，其特征在于，所述基于所述第一光源信息和所述第二光源信息，确定影响限制信息，包括：

基于所述第一光源信息，确定第一转换矩阵，其中，所述第一转换矩阵用于将所述待处理图像转换为所述待处理图像的白平衡处理后图像；

基于所述第二光源信息，确定第二转换矩阵，其中，所述第二转换矩阵用于将所述转换后图像转换为所述转换后图像的白平衡处理后图像；

基于所述第一转换矩阵和所述第二转换矩阵，确定影响限制信息。

3.根据权利要求2所述的方法，其特征在于，所述将所述待处理图像的色彩空间转换为第二色彩空间，包括：

采用第三转换矩阵，将所述待处理图像的色彩空间转换为第二色彩空间，其中，所述第三转换矩阵用于将所述第一色彩空间下的图像转换为所述第二色彩空间下的图像。

4.根据权利要求 3所述的方法，其特征在于，所述第一转换矩阵、所述第二转换矩阵和所述第三转换矩阵为对角矩阵。

5.根据权利要求1-3之一所述的方法，其特征在于，所述第一色彩空间为国际照明委员会建立的标准色度观察者色彩空间CIEXYZ，或者，图像信号处理流程色彩空间转换之前的摄像头自身的红绿蓝色彩空间native RGB；所述第二色彩空间为标准红绿蓝色彩空间sRGB。

6.一种图像的白平衡处理装置，其特征在于，所述装置包括：

获取单元，被配置成获取第一色彩空间下的待处理图像；

转换单元，被配置成将所述待处理图像的色彩空间转换为第二色彩空间，得到转换后图像；

确定单元，被配置成基于所述待处理图像和所述转换后图像，确定影响限制信息，其中，所述影响限制信息用以限制色彩空间转换对白平衡处理的影响，所述影响限制信息基于对所述待处理图像进行光源估计得到的第一光源信息和对所述转换后图像进行光源估计得到的第二光源信息确定，或由预训练的影响限制信息确定模型基于所述待处理图像和所述转换后图像确定；

生成单元，被配置成基于所述影响限制信息，对所述待处理图像进行白平衡处理，生成白平衡处理后图像，包括：将所述待处理图像依次右乘所述影响限制信息、第一转换矩阵和第三转换矩阵，得到所述白平衡处理后图像，其中，所述第一转换矩阵用于将所述待处理图像转换为所述待处理图像的白平衡处理后图像，所述第三转换矩阵用于将所述第一色彩空间下的图像转换为所述第二色彩空间下的图像。

7.一种电子设备，其特征在于，包括：

存储器，用于存储计算机程序；

处理器，用于执行所述存储器中存储的计算机程序，且所述计算机程序被执行时，实现上述权利要求1-5任一所述的方法。

8.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该计算机程序被处理器执行时，实现上述权利要求1-5任一所述的方法。