[发明专利]一种图像的白平衡校正方法、装置、存储介质及设备

权利要求书

1.一种图像的白平衡校正方法，其特征在于，该方法包括：

获取原始图像，对所述原始图像进行亮度范围预处理，得到像素权重值；

根据所述像素权重值确定原始图像的像素点在颜色空间中各颜色通道的平均值，根据所述各颜色通道的平均值构成平均值向量；

根据所述平均值向量确定原始图像的亮点集合和暗点集合，并根据所述亮点集合和暗点集合确定光照颜色；

根据所述光照颜色，对所述原始图像进行白平衡校正。

2.根据权利要求1所述的方法，其特征在于，对所述原始图像进行亮度范围预处理，得到像素权重值，包括：

获取原始图像中当前像素点在三个通道的亮度值最小值；

若所述亮度值最小值小于或者等于预设阈值，则确定该像素点的像素权重值为所述亮度值最小值；若所述亮度值最小值大于预设阈值，则确定该像素点的像素权重值为0；

遍历所述原始图像的所有像素点，得到原始图像的像素权重值。

3.根据权利要求1所述的方法，其特征在于，根据所述像素权重值确定原始图像的像素点在颜色空间中各颜色通道的平均值，包括：

采用如下公式分别计算原始图像的像素点在颜色空间中各颜色通道的平均值：

其中，R_avg为红颜色通道的平均值，G_avg为绿颜色通道的平均值，B_avg为蓝颜色通道的平均值，ω(x,y)为为坐标是(x，y)的像素点的像素权重值，f^R(x,y)为坐标是(x，y)的像素点的红颜色通道的亮度值，f^G(x,y)为坐标是(x，y)的像素点的绿颜色通道的亮度值，f^B(x,y)为坐标是(x，y)的像素点的蓝颜色通道的亮度值。

4.根据权利要求1所述的方法，其特征在于，根据所述平均值向量确定原始图像的亮点集合和暗点集合，并根据所述亮点集合和暗点集合确定光照颜色，包括：

将原始图像的像素点投影到所述平均值向量上，得到各像素点的投影长度；

根据所述投影长度确定原始图像的亮点集合和暗点集合。

5.根据权利要求4所述的方法，其特征在于，根据所述投影长度确定原始图像的亮点集合和暗点集合，包括：

获取预设亮点集合选取规则和暗点集合选取规则；

按照所述亮点集合选取规则对各像素点的投影长度进行筛选，确定亮点集合；以及，按照所述暗点集合选取规则对各像素点的投影长度进行筛选，确定暗点集合。

6.根据权利要求1所述的方法，其特征在于，根据所述亮点集合和暗点集合确定光照颜色，包括：

根据所述亮点集合和暗点集合，确定二维矩阵；其中，所述二维矩阵的行数为亮点集合和暗点集合的像素点总数，列数为三列，分别为亮点集合和暗点集合的像素点的三个颜色通道亮度值；

根据所述二维矩阵，生成协方差矩阵；

根据所述协方差矩阵，确定候选特征值与候选特征向量；其中，所述候选特征值与候选特征向量一一对应，所述候选特征值的数量为至少一个；

确定所述候选特征值最大的一个所对应的候选特征向量的向量坐标值为光照颜色。

7.根据权利要求1所述的方法，其特征在于，根据所述光照颜色，对所述原始图像进行白平衡校正，包括：

根据所述光照颜色，确定各颜色通道的颜色增益；

根据所述颜色增益，与原始图像的各颜色通道的像素点亮度值，确定原始图像的白平衡矫正结果。