[发明专利]图像颜色风格转移方法

说明书

本发明公开了一种图像颜色风格转移方法，包括以下步骤：对参考图像和源图像预处理；通过卷积神经网络得到预处理后的参考图像和源图像的特征图；获取参考图像的特征图和源图像的特征图的相互映射函数；根据相互映射函数获取颜色转移导向图；获取缩小分辨率后的源图像到颜色转移导向图的颜色转移函数；根据颜色转移函数对缩小分辨率后的源图像进行颜色转移得到第一目标图像，调节第一目标图像的分辨率得到第二目标图像；将第二目标图像作为新的源图像，重复执行上述步骤一定次数得到最终图像。图像颜色风格转移方法利用卷积神经网络进行深度匹配建立两个图像之间的对应关系用于指导精确的颜色转换，并且通过重复迭代的方式提高颜色转换的准确性。

技术领域

本发明涉及一种图像颜色风格转移方法。

背景技术

目前颜色转移是一个长期存在的问题，我们试图转移参考照片的颜色风格到另一个图像上，通过选取适当的参考照片，可以使输入图像看起来像是在不同的时间、天气条件、照明下拍摄的。该方法可用于艺术颜色修饰的效果。到目前为止，现有技术不可避免地面临一些挑战。

发明内容

本发明提供了一种图像颜色风格转移方法，采用如下的技术方案：

一种图像颜色风格转移方法，用于将参考图像的颜色风格转移至源图像，图像颜色风格转移方法包括以下步骤：

S1：对参考图像进行预处理；

S2：对源图像进行预处理；

S3：将经过步骤S1和步骤S2得到的预处理后的参考图像和源图像分别输入到预先训练好的卷积神经网络得到预处理后的参考图像的特征图和预处理后的源图像的特征图；

S4：获取参考图像的特征图到源图像的特征图的第一映射函数以及源图像的特征图到参考图像的特征图的第二映射函数；

S5：根据第一映射函数和第二映射函数获取颜色转移导向图；

S6：将源图像的分辨率缩小以与颜色转移导向图的分辨率相同；

S7：获取缩小分辨率后的源图像到颜色转移导向图的颜色转移函数；

S8：根据颜色转移函数对缩小分辨率后的源图像进行颜色转移得到第一目标图像，根据源图像的分辨率调节第一目标图像的分辨率得到第二目标图像。

S9：将第二目标图像作为新的源图像，重复执行步骤S2到步骤S8一定次数，得到最终图像。

进一步地，步骤S8进一步为：根据颜色转移函数通过局部颜色转移法对缩小分辨率后的源图像进行颜色转移得到第一目标图像，根据源图像的分辨率调节第一目标图像的分辨率得到第二目标图像。

进一步地，步骤S8进一步为：根据颜色转移函数通过局部颜色转移法对缩小分辨率后的源图像进行颜色转移得到第一目标图像，在进行颜色转移的过程中强制执行局部平滑和非局部约束，根据源图像的分辨率调节第一目标图像的分辨率得到第二目标图像。

进一步地，步骤S9进一步为：将第二目标图像作为新的源图像，重复执行步骤S2到S8大于等于3次且小于等于6次，得到最终图像。

进一步地，步骤S9进一步为：将第二目标图像作为新的源图像，重复执行步骤S2到S8达到6次，得到最终图像。

进一步地，步骤S1和步骤S2中，对参考图像和源图像进行预处理的具体方法为：

对参考图像和源图像进行高通滤波。

进一步地，对参考图像进行预处理的具体方法还包括：

对参考图像进行裁剪。

进一步地，步骤S4进一步为：通过最近邻域滤波算法获取参考图像的特征图到源图像的特征图的第一映射函数以及源图像的特征图到参考图像的特征图的第二映射函数。