[发明专利]一种图像色彩迁移方法、系统及计算机介质

权利要求书

1.一种图像色彩迁移方法，其特征在于，包括以下步骤：

S1：通过色彩空间的转换矩阵将源图像S和目标图像T的RGB色彩空间转换到Lab色彩空间；

S2：根据转换后的源图像S和目标图像T分别进行数据集的建立；

S3：建立第一KNN回归模型和第二KNN回归模型，利用所述数据集分别训练所述第一KNN回归模型和第二KNN回归模型，得到训练好的从通道L映射到通道a的第一KNN回归模型和从通道L映射到通道b的第二KNN回归模型；

S4：利用训练好的第一KNN回归模型和第二KNN回归模型，分别预测源图像S中通道L映射的通道a和通道b的值，得到映射后的图像；

S5：将映射后的图像转换回RGB色彩空间，得到结果图像，实现将目标图像T的色彩迁移到源图像S上。

2.根据权利要求1所述的图像色彩迁移方法，其特征在于，步骤S2中根据转换后的源图像S和目标图像T分别进行数据集的建立，具体为：

所述数据集包括训练集和预测集，所述训练集为从目标图像T的通道L中的N×N矩形块和与所述N×N矩形块的中心点所对应的通道a和通道b的值，所述模型的预测集为源图像S的通道L中N×N的矩形块。

3.根据权利要求2所述的图像色彩迁移方法，其特征在于，所述第一KNN回归模型和第二KNN回归模型，具体为：

在目标图像T的通道L中，对于每一个N×N矩形块，找到K个与所述N×N矩形块数值距离最近的块，若数值距离越相近，则两个矩形块中在通道L的信息也越相似，为了使数值距离较近的矩形块对选定矩形块的影响大于数值距离较远的矩形块的影响，用数值距离的倒数来进行加权，从通道L映射到通道a的第一KNN回归模型和从通道L映射到通道b的第二KNN回归模型。

4.根据权利要求3所述的图像色彩迁移方法，其特征在于，所述数值距离的度量方式为曼哈顿距离。

5.根据权利要求4所述的图像色彩迁移方法，其特征在于，所述N×N矩形块具体为3×3矩形块。

6.根据权利要求1至5任一项所述的图像色彩迁移方法，其特征在于，在步骤S1将源图像S的RGB色彩空间转换到Lab色彩空间前，还包括步骤：

利用Reinhard色彩迁移算法对源图像S进行一次Reinhard色彩迁移操作，所述的源图像S为彩色图像。

7.根据权利要求1至5任一项所述的图像色彩迁移方法，其特征在于，在步骤S5得到结果图像后，还包括步骤：

利用Reinhard色彩迁移算法对结果图像进行一次Reinhard色彩迁移操作。

8.一种图像色彩迁移系统，其特征在于，包括：

色彩空间转换模块，所述色彩空间转换模块用于通过色彩空间的转换矩阵将源图像S和目标图像T的RGB色彩空间转换到Lab色彩空间；

数据集建立模块，所述数据集建立模块用于根据转换后的源图像S和目标图像T分别进行数据集的建立；

回归模型模块，所述回归模型模块用于建立第一KNN回归模型和第二KNN回归模型，利用所述数据集分别训练所述第一KNN回归模型和第二KNN回归模型，得到训练好的从通道L映射到通道a的第一KNN回归模型和从通道L映射到通道b的第二KNN回归模型；

预测模块，所述预测模块用于利用训练好的第一KNN回归模型和第二KNN回归模型，分别预测源图像S中通道L映射的通道a和通道b的值，得到映射后的图像；

输出模块，所述输出模块用于将映射后的图像转换回RGB色彩空间，得到结果图像，实现将目标图像T的色彩迁移到源图像S上。