[发明专利]光源估计方法、装置、电子设备、芯片及存储介质

说明书

本公开涉及计算机视觉技术领域，具体提供了一种光源估计方法、装置、电子设备、芯片、及存储介质。一种光源估计方法，包括：获取当前场景光源的多光谱数据；根据预先设置的标准光源数据集和多光谱数据，确定标准光源数据集对应的第一权重集；其中，标准光源数据集包括至少两个标准光源的光谱数据，第一权重集包括每个标准光源对应的第一权值；根据标准光源数据集中各标准光源的光谱数据和第一权值，确定当前场景光源的估计结果。本公开实施方式中，提高光源估计准确性，图像色彩还原与人眼实际感知到的色彩更为接近，提高成像效果。

技术领域

本公开涉及图像处理技术领域，具体涉及一种光源估计方法、装置、电子设备、芯片及存储介质。

背景技术

对于人体视觉系统，在场景光源变化的情况下，人眼能够不受场景光源变化的影响而感知到物体自身固有的颜色，这种能力被称为颜色恒常性。而对于计算机视觉系统，在场景光源发生变化的时候，图像传感器的成像效果会出现不同程度的色偏，例如物体偏蓝、偏黄等。

光源估计算法可以对当前场景光源的相关色温或白点坐标进行估计，从而将场景图像恢复至标准光源下，实现图像的色彩还原。但是，相关技术中，光源估计效果较差，导致最终还原的图像色彩与人眼实际感知存在偏差，图像色彩效果不佳。

发明内容

为提高光源估计准确性，本公开实施方式提供了一种光源估计方法、装置、电子设备、芯片及存储介质。

第一方面，本公开实施方式提供了一种光源估计方法，包括：

获取当前场景光源的多光谱数据；

根据预先设置的标准光源数据集和所述多光谱数据，确定所述标准光源数据集对应的第一权重集；其中，所述标准光源数据集包括至少两个标准光源的光谱数据，所述第一权重集包括每个标准光源对应的第一权值；

根据所述标准光源数据集中各标准光源的所述光谱数据和所述第一权值，确定所述当前场景光源的估计结果。

在一些实施方式中，所述根据预先设置的标准光源数据集和所述多光谱数据，确定所述标准光源数据集对应的第一权重集，包括：

根据所述标准光源数据集和所述多光谱数据，对每个标准光源的初始权值进行回归处理，得到每个标准光源对应的第一权值；

根据每个标准光源对应的第一权值，确定所述第一权重集。

在一些实施方式中，所述标准光源数据集包括每个标准光源的相关色温数据；所述根据所述标准光源数据集中各标准光源的所述光谱数据和所述第一权值，确定所述当前场景光源的估计结果，包括：

根据所述第一权重集中的各个的所述第一权值，从所述标准光源数据集中确定至少两个候选标准光源；

根据每个候选标准光源的所述光谱数据和所述相关色温数据、以及所述多光谱数据，确定当前场景光源的估计结果。

在一些实施方式中，所述根据各个所述标准光源的所述第一权值，从所述标准光源数据集中确定至少两个候选标准光源，包括：

对所述第一权重集中各个所述第一权值由大至小进行排序，得到第一排序集合；

根据所述第一排序集合中的最大值与预设权重阈值的关系，将排序前预设数量的第一权值对应的标准光源确定为所述候选标准光源。

在一些实施方式中，所述根据所述第一排序集合中的最大值与预设权重阈值的关系，将排序前预设数量的第一权值对应的标准光源确定为所述候选标准光源，包括：

响应于所述第一排序集合中的最大值不小于所述预设权重阈值，将排序前第一数量的第一权值对应的标准光源确定为所述候选标准光源；

和/或，