[发明专利]光源光谱获取方法和设备

权利要求书

1.一种光源光谱获取方法，其特征在于，所述方法包括：

获取当前拍摄场景下的第一信息，所述第一信息包括以下至少一种：RGB传感器生成的第一图像或者第一多光谱传感器上各个像素接收到的光线的光强度；

将所述第一信息输入第一模型，得到当前拍摄场景下的光源属于各类光源的概率；

根据当前拍摄场景下的光源属于各类光源的概率以及各类光源的光谱，确定当前拍摄场景下的光源的光谱。

2.根据权利要求1所述的方法，其特征在于，所述将所述第一信息输入第一模型，得到当前拍摄场景下的光源属于各类光源的概率之前，所述方法还包括：

获取训练样本，所述训练样本包括不同拍摄场景下的第一信息以及不同拍摄场景下的光源对应的光源类别；

使用所述训练样本训练所述第一模型。

3.根据权利要求2所述的方法，其特征在于，所述获取训练样本，包括：

构建数据集，所述数据集包括M×N组数据，每组数据对应一种拍摄场景，每种拍摄场景对应一种光源和一种被摄物，M为光源的数量，N为被摄物数量，每组数据包括：对应拍摄场景下的第一信息；

对所述数据集中M种光源的光谱进行聚类分析，得到每种拍摄场景下的光源对应的光源类别；

根据每组数据中的第一信息以及每种拍摄场景下的光源对应的光源类别，获取训练样本。

4.根据权利要求3所述的方法，其特征在于，所述M×N组数据中每组数据还包括：第二信息，所述第二信息包括：对应拍摄场景下的光源的光谱以及对应拍摄场景下第二多光谱传感器上各个像素对应的物体材质反射率谱；

所述对所述数据集中M种光源的光谱进行聚类分析，得到每种拍摄场景下的光源对应的光源类别，包括：

获取每组数据中的第二信息，得到M×N组第二信息；

根据M×N组第二信息，对所述数据集中M种光源的光谱进行聚类分析。

5.根据权利要求4所述的方法，其特征在于，所述根据M×N组第二信息，对所述数据集中M种光源的光谱进行聚类分析，包括：

根据M×N组第二信息，确定所述数据集中M种光源的分类标准；

根据所述分类标准，对所述数据集中M种光源的光谱进行聚类分析。

6.根据权利要求5所述的方法，其特征在于，所述根据M×N组第二信息，确定所述数据集中M种光源的分类标准，包括：

针对N种被摄物中的每种被摄物，从所述被摄物对应的M组第二信息中任取两组，得到对组合；

针对每对组合，计算所述组合对应的二维数据点，得到个二维数据点，所述二维数据点包括平均颜色距离和光源光谱相似指标；

根据所述个二维数据点，确定所述分类标准。

7.根据权利要求6所述的方法，其特征在于，所述针对每对组合，计算所述组合对应的二维数据点，包括：

根据所述组合中一组第二信息，计算各个像素的第一RGB值；

根据所述组合中另一组第二信息，计算各个像素的第二RGB值；

根据各个像素的第一RGB值和各个像素的第二RGB值，确定所述组合对应的平均颜色距离。

8.根据权利要求6所述的方法，其特征在于，所述针对每对组合，计算所述组合对应的二维数据点，包括：

根据所述组合中一组第二信息包含的光源的光谱和所述组合中另一组第二信息中包含的光源的光谱，确定所述组合对应的光源光谱相似指标。

9.根据权利要求6-8任一项所述的方法，其特征在于，所述根据所述个二维数据点，确定所述分类标准，包括：

根据所述个二维数据点，绘制二维交会图；

从所述二维交会图的横坐标上确定第一参数，使得纵坐标小于预设值，且横坐标小于所述第一参数的二维点数据点的数量所占的比例超过预设阈值；

将所述第一参数作为所述分类标准。