[发明专利]用于校正模型训练和图像校正的方法和装置及电子设备

权利要求书

1.一种色彩校正模型训练方法，包括：

获取色卡的图像；

使用色卡的图像来训练神经网络，从而获得第一色彩校正模型；

利用所述第一色彩校正模型对色彩空间中的任意原始原色值进行校正，以获得对应的目标原色值；以及

使用所述任意原始原色值作为输入数据，并且使用对应的目标原色值作为监督数据，来训练第二色彩校正模型，

其中，所述第二色彩校正模型是基于非神经网络的色彩校正模型，且

其中，所述色卡的图像包括不同色温下的图像，从而所述第一色彩校正模型和所述第二色彩校正模型基于色温来执行色彩校正。

2.如权利要求1所述的方法，其中，所述第一色彩校正模型是反向传播神经网络模型，所述第二色彩校正模型是线性回归模型。

3.如权利要求1所述的方法，其中，训练第二色彩校正模型包括：

将所述色彩空间中的任意原始原色值和对应的目标原色值按范围分为多个区块；

分别使用所述多个区块来训练第二色彩校正模型，以获得分别与所述多个区块对应的多个第二色彩校正子模型，所述第二色彩校正模型由多个第二色彩校正子模型组成。

4.如权利要求3所述的方法，其中，将所述色彩空间中的任意原始原色值和对应的目标原色值按范围分为多个区块包括：在所述色彩空间的每个维度上进行均匀分块。

5.如权利要求1所述的方法，其中，所述色彩空间包括RGB、YUV、HSV、Lab、CMYK中的任意一个。

6.一种图像色彩校正方法，包括：

接收待校正图像；以及

使用权利要求1-5中的任一项所述的第二色彩校正模型来校正所述待校正图像。

7.如权利要求6所述的方法，其中，所述第二色彩校正模型由多个第二色彩校正子模型组成，所述多个第二色彩校正子模型分别用于校正不同数值范围内的原色值，且

其中，所述图像色彩校正方法还包括根据所述待校正图像的原色值所属的数值范围，选择相对应的第二色彩校正子模型来进行校正。

8.一种色彩校正模型训练装置，包括：

图像获取单元，用于获取色卡的图像；

第一训练单元，用于使用所述色卡的图像来训练神经网络，从而获得第一色彩校正模型；

数据增强单元，用于利用所述第一色彩校正模型对色彩空间中的任意原始原色值进行校正，以获得对应的目标原色值；以及

第二训练单元，用于使用所述任意原始原色值作为输入数据，并且使用对应的目标原色值作为监督数据，来训练第二色彩校正模型，

其中，所述第二色彩校正模型是基于非神经网络的色彩校正模型，且

其中，所述色卡的图像包括不同色温下的图像，从而所述第一色彩校正模型和所述第二色彩校正模型基于色温来执行色彩校正。

9.如权利要求8所述的装置，还包括：

分块单元，用于将所述色彩空间中的任意原始原色值按范围分为多个区块，

其中，所述数据增强单元利用所述第一色彩校正模型对各个区块的原始原色值进行校正，以获得对应的各个区块的目标原色值，且

其中，所述第二训练单元分别使用各个区块的原始原色值和目标原色值来训练第二色彩校正模型，以获得分别与各个区块对应的多个第二色彩校正子模型，所述第二色彩校正模型由多个第二色彩校正子模型组成。

10.一种图像色彩校正装置，包括：

接收单元，用于接收待校正图像；以及

校正单元，用于使用权利要求1-5中的任一项所述的第二色彩校正模型来校正所述待校正图像。