[发明专利]图像颜色校正方法、装置、存储介质及移动终端

权利要求书

1.一种图像颜色校正方法，其特征在于，包括：

获取待处理的原始图像；

将所述原始图像输入至预先训练的图像颜色校正模型中；

确定所述图像颜色校正模型的输出图像，并将所述输出图像作为与所述原始图像对应的目标图像；

其中，在获取待处理的原始图像之前，还包括：

获取所述图像颜色校正模型；

其中，所述图像颜色校正模型由如下方式得到：

通过摄像头采集至少两个拍摄场景在不同光照下的第一样本原始图像；

对所述第一样本原始图像进行颜色校正，得到与所述第一样本原始图像对应的第一样本目标图像；

将所述第一样本原始图像和所述第一样本目标图像作为第一训练样本集；

利用所述第一训练样本集对第一预设机器学习模型进行训练，得到图像颜色校正模型；

其中，在将所述原始图像输入至预先训练的图像颜色校正模型中之前，还包括：

将所述原始图像输入至预先训练的白平衡系数矩阵确定模型中；

根据所述白平衡系数矩阵确定模型的输出结果，确定与所述原始图像对应的白平衡系数矩阵；

根据所述白平衡系数矩阵对所述原始图像进行白平衡处理；

所述将所述原始图像输入至预先训练的图像颜色校正模型中，包括：

将经白平衡处理后的原始图像输入至预先训练的图像颜色校正模型中；

其中，根据所述白平衡系数矩阵对所述原始图像进行白平衡处理，包括：获取所述原始图像中每个像素点的第一RGB分量值；针对所述原始图像中所有像素点，将每个像素点的第一RGB分量值与所述白平衡系数矩阵中对应位置的白平衡系数的乘积，作为与原始图像所述像素点对应的目标图像的像素点的第二RGB分量值。

2.根据权利要求1所述的方法，其特征在于，在对所述第一样本原始图像进行颜色校正，得到与所述第一样本原始图像对应的第一样本目标图像之前，还包括：

通过摄像头采集与所述第一样本原始图像对应的样本RGB图像；

对所述第一样本原始图像进行颜色校正，得到与所述第一样本原始图像对应的第一样本目标图像，包括：

以所述样本RGB图像为参考图像，对所述第一样本原始图像进行颜色校正，得到与所述第一样本原始图像对应的第一样本目标图像。

3.根据权利要求1所述的方法，其特征在于，在将所述原始图像输入至预先训练的白平衡系数矩阵确定模型中之前，还包括：

获取白平衡系数矩阵确定模型；

其中，所述白平衡系数矩阵确定模型由如下方式得到：

通过摄像头采集第二标准色卡在不同色温下的第二样本原始图像；其中，所述第二标准色卡为白色色卡；

对所述第二样本原始图像进行白平衡处理，得到与所述第二样本原始图像对应的第二样本目标图像；

根据所述第二样本原始图像和所述第二样本目标图像，确定将所述第二样本原始图像变化为所述第二样本目标图像对应的样本白平衡系数矩阵；

根据所述样本白平衡系数矩阵对所述第二样本原始图像进行标记，得到第二训练样本集；

利用所述第二训练样本集对第二预设机器学习模型进行训练，得到白平衡系数矩阵确定模型。

4.根据权利要求1-3任一所述的方法，其特征在于，在将所述输出图像作为与所述原始图像对应的目标图像之后，还包括：

对所述目标图像进行Gamma校正，并输出Gamma校正后的目标图像。