本公开涉及计算机视觉技术领域，具体提供了一种光源估计方法、装置、电子设备、芯片、及存储介质。一种光源估计方法，包括：获取当前场景光源的多光谱数据；根据预先设置的标准光源数据集和多光谱数据，确定标准光源数据集对应的第一权重集；其中，标准光源数据集包括至少两个标准光源的光谱数据，第一权重集包括每个标准光源对应的第一权值；根据标准光源数据集中各标准光源的光谱数据和第一权值，确定当前场景光源的估计结果。本公开实施方式中，提高光源估计准确性，图像色彩还原与人眼实际感知到的色彩更为接近，提高成像效果。提高成像效果。提高成像效果。
【技术实现步骤摘要】
光源估计方法、装置、电子设备、芯片及存储介质
[0001]本公开涉及图像处理
，具体涉及一种光源估计方法、装置、电子设备、芯片及存储介质。
技术介绍
[0002]对于人体视觉系统，在场景光源变化的情况下，人眼能够不受场景光源变化的影响而感知到物体自身固有的颜色，这种能力被称为颜色恒常性。而对于计算机视觉系统，在场景光源发生变化的时候，图像传感器的成像效果会出现不同程度的色偏，例如物体偏蓝、偏黄等。
[0003]光源估计算法可以对当前场景光源的相关色温或白点坐标进行估计，从而将场景图像恢复至标准光源下，实现图像的色彩还原。但是，相关技术中，光源估计效果较差，导致最终还原的图像色彩与人眼实际感知存在偏差，图像色彩效果不佳。
技术实现思路
[0004]为提高光源估计准确性，本公开实施方式提供了一种光源估计方法、装置、电子设备、芯片及存储介质。
[0005]第一方面，本公开实施方式提供了一种光源估计方法，包括：
[0006]获取当前场景光源的多光谱数据；
[0007]根据预先设置的标准光源数据集和所述多光谱数据，确定所述标准光源数据集对应的第一权重集；其中，所述标准光源数据集包括至少两个标准光源的光谱数据，所述第一权重集包括每个标准光源对应的第一权值；
[0008]根据所述标准光源数据集中各标准光源的所述光谱数据和所述第一权值，确定所述当前场景光源的估计结果。
[0009]在一些实施方式中，所述根据预先设置的标准光源数据集和所述多光谱数据，确定所述标准光源数据集对应的第一权重集，包括：
[0010]根据所述标准光源数据集和所述多光谱数据，对每个标准光源的初始权值进行回归处理，得到每个标准光源对应的第一权值；
[0011]根据每个标准光源对应的第一权值，确定所述第一权重集。
[0012]在一些实施方式中，所述标准光源数据集包括每个标准光源的相关色温数据；所述根据所述标准光源数据集中各标准光源的所述光谱数据和所述第一权值，确定所述当前场景光源的估计结果，包括：
[0013]根据所述第一权重集中的各个的所述第一权值，从所述标准光源数据集中确定至少两个候选标准光源；
[0014]根据每个候选标准光源的所述光谱数据和所述相关色温数据、以及所述多光谱数据，确定当前场景光源的估计结果。
[0015]在一些实施方式中，所述根据各个所述标准光源的所述第一权值，从所述标准光
源数据集中确定至少两个候选标准光源，包括：
[0016]对所述第一权重集中各个所述第一权值由大至小进行排序，得到第一排序集合；
[0017]根据所述第一排序集合中的最大值与预设权重阈值的关系，将排序前预设数量的第一权值对应的标准光源确定为所述候选标准光源。
[0018]在一些实施方式中，所述根据所述第一排序集合中的最大值与预设权重阈值的关系，将排序前预设数量的第一权值对应的标准光源确定为所述候选标准光源，包括：
[0019]响应于所述第一排序集合中的最大值不小于所述预设权重阈值，将排序前第一数量的第一权值对应的标准光源确定为所述候选标准光源；
[0020]和/或，
[0021]响应于所述第一排序集合中的最大值小于所述预设权重阈值，将排序前第二数量的第一权值对应的标准光源确定为所述候选标准光源；其中，所述第二数量大于所述第一数量。
[0022]在一些实施方式中，所述根据每个候选标准光源的所述光谱数据和所述相关色温数据，以及所述多光谱数据，确定当前场景光源的估计结果，包括：
[0023]根据每个候选标准光源的所述光谱数据和所述相关色温数据、以及所述多光谱数据，对每个候选标准光源的第一权值进行回归处理，得到每个候选标准光源的第二权值；
[0024]根据所述第二权值，从所述候选标准光源中确定目标数量的目标标准光源；
[0025]根据各个所述目标标准光源的光谱数据和第二权值，确定当前场景光源的估计结果。
[0026]在一些实施方式中，在得到所述当前场景光源的估计结果之后，还包括：
[0027]根据所述当前场景光源的估计结果和所述多光谱数据，确定所述估计结果的置信度分值。
[0028]在一些实施方式中，预先设置标准光源数据集的过程包括：
[0029]通过多光谱传感器采集所述至少两个标准光源的初始光谱数据，并根据所述初始光谱数据得到每个标准光源的光谱曲线；
[0030]对于每个标准光源，根据所述光谱曲线提取靠近多光谱传感器响应峰值点的预设波段的光谱数据，作为所述标准光源的光谱数据；
[0031]根据所有标准光源的光谱数据，得到所述标准光源数据集。
[0032]第二方面，本公开实施方式提供了一种光源估计装置，包括：
[0033]获取模块，被配置为获取当前场景光源的多光谱数据；
[0034]权重确定模块，被配置为根据预先设置的标准光源数据集和所述多光谱数据，确定所述标准光源数据集对应的第一权重集；其中，所述标准光源数据集包括至少两个标准光源的光谱数据，所述第一权重集包括每个标准光源对应的第一权值；
[0035]结果确定模块，被配置为根据所述标准光源数据集中各标准光源的所述光谱数据和所述第一权值，确定所述当前场景光源的估计结果。
[0036]在一些实施方式中，所述权重确定模块被配置为：
[0037]根据所述标准光源数据集和所述多光谱数据，对每个标准光源的初始权值进行回归处理，得到每个标准光源对应的第一权值；
[0038]根据每个标准光源对应的第一权值，确定所述第一权重集。
[0039]在一些实施方式中，所述标准光源数据集包括每个标准光源的相关色温数据；所述结果确定模块，被配置为：
[0040]根据所述第一权重集中的各个的所述第一权值，从所述标准光源数据集中确定至少两个候选标准光源；
[0041]根据每个候选标准光源的所述光谱数据和所述相关色温数据、以及所述多光谱数据，确定当前场景光源的估计结果。
[0042]在一些实施方式中，所述结果确定模块被配置为：
[0043]对所述第一权重集中各个所述第一权值由大至小进行排序，得到第一排序集合；
[0044]根据所述第一排序集合中的最大值与预设权重阈值的关系，将排序前预设数量的第一权值对应的标准光源确定为所述候选标准光源。
[0045]在一些实施方式中，所述结果确定模块被配置为：
[0046]响应于所述第一排序集合中的最大值不小于所述预设权重阈值，将排序前第一数量的第一权值对应的标准光源确定为所述候选标准光源；
[0047]和/或，
[0048]响应于所述第一排序集合中的最大值小于所述预设权重阈值，将排序前第二数量的第一权值对应的标准光源确定为所述候选标准光源；其中，所述第二数量大于所述第一数量。
[0049]在一些实施方式中，所述结果确定模块被配置为：
[0050]根据每个候选标准光源本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种光源估计方法，其特征在于，包括：获取当前场景光源的多光谱数据；根据预先设置的标准光源数据集和所述多光谱数据，确定所述标准光源数据集对应的第一权重集；其中，所述标准光源数据集包括至少两个标准光源的光谱数据，所述第一权重集包括每个标准光源对应的第一权值；根据所述标准光源数据集中各标准光源的所述光谱数据和所述第一权值，确定所述当前场景光源的估计结果。2.根据权利要求1所述的方法，其特征在于，所述根据预先设置的标准光源数据集和所述多光谱数据，确定所述标准光源数据集对应的第一权重集，包括：根据所述标准光源数据集和所述多光谱数据，对每个标准光源的初始权值进行回归处理，得到每个标准光源对应的第一权值；根据每个标准光源对应的第一权值，确定所述第一权重集。3.根据权利要求1或2所述的方法，其特征在于，所述标准光源数据集包括每个标准光源的相关色温数据；所述根据所述标准光源数据集中各标准光源的所述光谱数据和所述第一权值，确定所述当前场景光源的估计结果，包括：根据所述第一权重集中的各个的所述第一权值，从所述标准光源数据集中确定至少两个候选标准光源；根据每个候选标准光源的所述光谱数据和所述相关色温数据、以及所述多光谱数据，确定当前场景光源的估计结果。4.根据权利要求3所述的方法，其特征在于，所述根据各个所述标准光源的所述第一权值，从所述标准光源数据集中确定至少两个候选标准光源，包括：对所述第一权重集中各个所述第一权值由大至小进行排序，得到第一排序集合；根据所述第一排序集合中的最大值与预设权重阈值的关系，将排序前预设数量的第一权值对应的标准光源确定为所述候选标准光源。5.根据权利要求4所述的方法，其特征在于，所述根据所述第一排序集合中的最大值与预设权重阈值的关系，将排序前预设数量的第一权值对应的标准光源确定为所述候选标准光源，包括：响应于所述第一排序集合中的最大值不小于所述预设权重阈值，将排序前第一数量的第一权值对应的标准光源确定为所述候选标准光源；和/或，响应于所述第一排序集合中的最大值小于所述预设权重阈值，将排序前第二数量的第一权值对应的标准光源确定为所述候选标准光源；其中，所述第二数量大于所述第一数量。6.根据权利要求3所述的方法，其特征在于，所述根据每...
【专利技术属性】
技术研发人员：张雪岩，
申请(专利权)人：上海玄戒技术有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。