[发明专利]一种色彩校正方法、装置、计算机设备及存储介质

权利要求书

1.一种色彩校正方法，其特征在于，所述方法包括以下步骤：

标定标准光源下的色彩矩阵，所述标准光源下的色彩矩阵用于表征标准色卡中每个色块的标准值和真实值之间的关系；

利用带权重的梯度下降法对标准光源下的色彩矩阵进行迭代优化，得到优化后的标准光源下的色彩矩阵；

计算实际环境光源下的色彩矩阵；

将得到的实际环境光源下的色彩矩阵作用到输入图像上逐像素点进行色彩校正。

2.根据权利要求1所述的方法，其特征在于，所述利用带权重的梯度下降法对标准光源下的色彩矩阵进行迭代优化，得到优化后的标准光源下的色彩矩阵的步骤，具体包括：

根据每个色块的R、G和B通道标准值分别建立矩阵；

根据每个色块的R、G和B通道真实值分别建立矩阵；

构建权重矩阵，所述权重矩阵中记录有每个色块对应的权重值；

根据权重矩阵和每个色块R、G和B通道标准值以及真实值建立得到的矩阵以及色彩矩阵的每一行参数分别建立损失函数；

分别对色彩矩阵的每一行参数做梯度下降法，直至迭代次数超过设定的第一阈值或损失函数低于设定的第二阈值，退出迭代，得到优化后的标准光源下的色彩矩阵的。

3.根据权利要求1或2所述的方法，其特征在于，所述利用带权重的梯度下降法对标准光源下的色彩矩阵进行迭代优化，得到优化后的标准光源下的色彩矩阵的步骤之后，还对优化后的标准光源下的色彩矩阵进行标准化操作，以剔除优化后的标准光源下的色彩矩阵对白平衡的影响。

4.根据权利要求1所述的方法，其特征在于，所述计算实际环境光源下的色彩矩阵的步骤，具体包括：

获取当前白平衡模块输出的RGB增益值；

计算当前白平衡模块输出的RGB增益值与每种标准光源的白平衡RGB增益值之间的距离，得到距离中的最小值和次最小值；

当所述距离的最小值小于设定的第三阈值时，以最小值对应的标准光源标定的色彩矩阵作为实际环境光源下的色彩矩阵；

当所述距离的最小值不小于设定的第三阈值时，则以最小值和次最小值分别对应的标准光源标定的色彩矩阵进行插值得到实际环境光源下的色彩矩阵。

5.根据权利要求1或4所述的方法，其特征在于，所述计算实际环境光源下的色彩矩阵之后，还对计算得到的实际环境光源下的色彩矩阵进行标准化操作。

6.一种色彩校正装置，其特征在于，所述装置包括：

色彩矩阵标定模块，用于标定标准光源下的色彩矩阵，所述标准光源下的色彩矩阵用于表征标准色卡中每个色块的标准值和真实值之间的关系；

迭代优化模块，用于利用带权重的梯度下降法对标准光源下的色彩矩阵进行迭代优化，得到优化后的标准光源下的色彩矩阵；

色彩矩阵计算模块，用于计算实际环境光源下的色彩矩阵；

色彩校正模块，用于将得到的实际环境光源下的色彩矩阵作用到输入图像上逐像素点进行色彩校正。

7.根据权利要求6所述的装置，其特征在于，所述迭代优化模块包括：

第一矩阵构建单元，用于根据每个色块的R、G和B通道标准值分别建立矩阵；

第二矩阵构建单元，用于根据每个色块的R、G和B通道真实值分别建立矩阵；

权重矩阵构建单元，用于构建权重矩阵，所述权重矩阵中记录有每个色块对应的权重值；

损失函数建立单元，用于根据权重矩阵和每个色块R、G和B通道标准值以及真实值建立得到的矩阵以及色彩矩阵的每一行参数分别建立损失函数；

梯度下降单元，用于分别对色彩矩阵的每一行参数做梯度下降法，直至迭代次数超过设定的第一阈值或损失函数低于设定的第二阈值，退出迭代，得到优化后的标准光源下的色彩矩阵。