[发明专利]源图像数据库的获取方法、训练装置及电子设备

权利要求书

1.一种源图像数据库的获取方法，所述源图像数据库用于训练神经网络，所述神经网络用于对采集单元采集的图像进行白平衡处理，其特征在于，所述获取方法包括步骤：

S100：获取已知图像库，并得到所述已知图像库中各原始图像对应的光源信息；其中，所述已知图像库中的所述原始图像覆盖多个色温点，且各所述原始图像中带有标准色卡；

S200：获取校正图像库，并计算所述校正图像库中各采集图像对应的光源信息；其中，所述校正图像库通过使用目标采集单元采集不同色温下的带标准色卡的采集图像形成；

S300：根据多组色温点相同的所述原始图像和所述采集图像中对应的光源信息，建立由所述原始图像转换到所述采集图像的色温模型，所述色温模型包括转换矩阵c和转换系数z₀，其中，所述原始图像的光源信息与所述转换矩阵c的乘积等于所述采集图像的光源信息与转换系数的乘积；

S400：根据所述色温模型，对所述已知图像库的各所述原始图像的光源信息进行转换，得到目标图像库中各目标图像的光源信息；

S500：根据所述已知图像库和所述目标图像库的光源信息将所述已知图像库的各所述原始图像的像素信息转换到所述目标图像库的各所述目标图像的像素信息，得到所述目标采集单元对应的源图像数据库；

其中，所述校正图像库的采集图像的个数小于所述已知图像库的原始图像的个数，且二者的图像来源不同；所述图像的光源信息包括所述图像的红色通道的亮度值l_r、绿色通道的亮度值l_g以及蓝色通道的亮度值l_b。

2.根据权利要求1所述的获取方法，其特征在于，所述步骤S100中获取已知图像库，具体包括：

通过使用非目标采集单元在不同的场景下采集带标准色卡的所述原始图像，形成所述已知图像库；或者

通过白平衡的公共数据集获取所述已知图像库。

3.根据权利要求1所述的获取方法，其特征在于，所述步骤S200中校正图像库通过使用目标采集单元采集不同色温下的带标准色卡的采集图像形成，具体包括：所述校正图像库通过使用目标采集单元在灯箱下采集不同标准光源的带标准色卡的采集图像形成。

4.根据权利要求1所述的获取方法，其特征在于，所述步骤S100中和所述步骤S200中得到光源信息的具体方法包括步骤：

S001：将图像转换到RGB域；

S002：分别对所述图像中标准色卡上未过曝的白块计算红色通道的像素均值、绿色通道的像素均值和蓝色通道的像素均值，三个均值分别为所述图像的红色通道的亮度值l_r、绿色通道的亮度值l_g以及蓝色通道的亮度值l_b；

其中，所述图像指所述原始图像或者采集图像。