光源估计方法、装置、电子设备、芯片及存储介质与流程

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

背景技术：

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

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

技术实现思路

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

17、和/或，

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

19、在一些实施方式中，所述根据每个候选标准光源的所述光谱数据和所述相关色温数据，以及所述多光谱数据，确定当前场景光源的估计结果，包括：

20、根据每个候选标准光源的所述光谱数据和所述相关色温数据、以及所述多光谱数据，对每个候选标准光源的第一权值进行回归处理，得到每个候选标准光源的第二权值；

21、根据所述第二权值，从所述候选标准光源中确定目标数量的目标标准光源；

22、根据各个所述目标标准光源的光谱数据和第二权值，确定当前场景光源的估计结果。

23、在一些实施方式中，在得到所述当前场景光源的估计结果之后，还包括：

24、根据所述当前场景光源的估计结果和所述多光谱数据，确定所述估计结果的置信度分值。

25、在一些实施方式中，预先设置标准光源数据集的过程包括：

26、通过多光谱传感器采集所述至少两个标准光源的初始光谱数据，并根据所述初始光谱数据得到每个标准光源的光谱曲线；

27、对于每个标准光源，根据所述光谱曲线提取靠近多光谱传感器响应峰值点的预设波段的光谱数据，作为所述标准光源的光谱数据；

28、根据所有标准光源的光谱数据，得到所述标准光源数据集。

29、第二方面，本公开实施方式提供了一种光源估计装置，包括：

30、获取模块，被配置为获取当前场景光源的多光谱数据；

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

32、结果确定模块，被配置为根据所述标准光源数据集中各标准光源的所述光谱数据和所述第一权值，确定所述当前场景光源的估计结果。

33、在一些实施方式中，所述权重确定模块被配置为：

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

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

36、在一些实施方式中，所述标准光源数据集包括每个标准光源的相关色温数据；所述结果确定模块，被配置为：

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

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

39、在一些实施方式中，所述结果确定模块被配置为：

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

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

42、在一些实施方式中，所述结果确定模块被配置为：

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

44、和/或，

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

46、在一些实施方式中，所述结果确定模块被配置为：

47、根据每个候选标准光源的所述光谱数据和所述相关色温数据、以及所述多光谱数据，对每个候选标准光源的第一权值进行回归处理，得到每个候选标准光源的第二权值；

48、根据所述第二权值，从所述候选标准光源中确定目标数量的目标标准光源；

49、根据各个所述目标标准光源的光谱数据和第二权值，确定当前场景光源的估计结果。

50、在一些实施方式中，所述结果确定模块被配置为：

51、根据所述当前场景光源的估计结果和所述多光谱数据，确定所述估计结果的置信度分值。

52、在一些实施方式中，本公开的光源估计装置，还包括数据集设置模块，所述数据集设置模块被配置为：

53、通过多光谱传感器采集所述至少两个标准光源的初始光谱数据，并根据所述初始光谱数据得到每个标准光源的光谱曲线；

54、对于每个标准光源，根据所述光谱曲线提取靠近多光谱传感器响应峰值点的预设波段的光谱数据，作为所述标准光源的光谱数据；

55、根据所有标准光源的光谱数据，得到所述标准光源数据集。

56、第三方面，本公开实施方式提供了一种电子设备，包括：

57、处理器；和

58、存储器，存储有计算机指令，所述计算机指令用于使所述处理器执行根据第一方面任意实施方式所述的方法。

59、第四方面，本公开实施方式提供了一种存储介质，存储有计算机指令，所述计算机指令用于使计算机执行根据第一方面任意实施方式所述的方法。

60、第五方面，本公开实施方式提供了一种芯片，包括一个或多个接口电路和一个或多个处理器；所述接口电路用于从电子设备的存储器接收信号，并向所述处理器发送所述信号，所述信号包括存储器中存储的计算机指令；当所述处理器执行所述计算机指令时，使得所述电子设备执行第一方面任意实施方式所述的方法。

61、本公开实施方式的光源估计方法，包括获取当前场景光源的多光谱数据，根据预先设置的标准光源数据集和多光谱数据，确定标准光源数据集对应的第一权重集，根据标准光源数据集中各个标准光源的光谱数据和第一权值，确定当前场景光源的估计结果。本公开实施方式中，利用标准光源的光谱数据确定当前场景光源的估计结果，可以准确地反映出当前场景光源的光谱类型，从而可以准确识别出光源的光谱差异性，提高色彩还原的准确性，使得到的图像与人眼实际感知到的色彩更为接近，提高成像效果。