[发明专利]图像色彩迁移方法及装置、计算机可读介质和电子设备

权利要求书

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

获取内容图像以及参考图像；

将所述内容图像和所述参考图像进行参数化处理，得到所述内容图像对应的第一统计特征和所述参考图像的第二统计特征；

对所述第一统计特征和所述第二统计特征进行迭代优化，得到所述内容图像和所述参考图像之间的色彩映射关系；

根据所述色彩映射关系对所述内容图像进行色彩迁移处理，得到与所述参考图像具有相似色彩分布特征的目标内容图像。

2.根据权利要求1所述的方法，其特征在于，所述统计特征包括概率分布；所述将所述内容图像和所述参考图像进行参数化处理，得到所述内容图像对应的第一统计特征和所述参考图像的第二统计特征，包括：

基于高斯混合模型对所述内容图像进行混合高斯拟合，得到所述内容图像的第一概率分布；以及

基于高斯混合模型对所述参考图像进行混合高斯拟合，得到所述参考图像的第二概率分布。

3.根据权利要求1所述的方法，其特征在于，对所述第一统计特征和所述第二统计特征进行迭代优化，得到所述内容图像和所述参考图像之间的色彩映射关系，包括：

构建分布拟合模型，所述分布拟合模型为卷积神经网络模型，所述分布拟合模型的损失函数为相对熵损失函数；

基于所述分布拟合模型对所述第一统计特征和所述第二统计特征进行非线性优化，以最小化所述相对熵损失函数，得到所述内容图像和所述参考图像之间的色彩映射关系。

4.根据权利要求1所述的方法，其特征在于，根据所述色彩映射关系对所述内容图像进行色彩迁移处理，得到与所述参考图像具有相似色彩分布特征的目标内容图像，包括：

根据所述色彩映射关系将所述内容图像中的原始像素值转化为目标像素值；

通过所述目标像素值生成与所述参考图像具有相似色彩分布特征的目标内容图像。

5.根据权利要求1所述的方法，其特征在于，所述方法还包括：

将所述内容图像、所述参考图像和所述目标内容图像作为样本图像集合；

基于所述样本图像集合对预构建的色彩迁移模型进行有监督训练，得到训练完成的色彩迁移模型。

6.根据权利要求5所述的方法，其特征在于，所述色彩迁移模型包括卷积层，所述卷积层包括1*1卷积核；所述方法还包括：

将所述内容图像和所述参考图像输入到所述预构建的色彩迁移模型中，以通过所述1*1卷积核分别提取所述内容图像中第一图像特征的和所述参考图像中的第二图像特征。

7.根据权利要求6所述的方法，其特征在于，所述色彩迁移模型包括自适应示例归一化模块；所述方法还包括：

通过所述自适应示例归一化模块将所述第一图像特征的均值和方差对齐到所述第二图像特征的均值和方差，以实现将所述参考图像中的色彩分布特征迁移到所述内容图像中。

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

图像获取模块，用于获取内容图像以及参考图像；

图像参数化模块，用于将所述内容图像和所述参考图像进行参数化处理，得到所述内容图像对应的第一统计特征和所述参考图像的第二统计特征；

优化拟合模块，用于对所述第一统计特征和所述第二统计特征进行迭代优化，得到所述内容图像和所述参考图像之间的色彩映射关系；

图像色彩迁移模块，用于根据所述色彩映射关系对所述内容图像进行色彩迁移处理，得到与所述参考图像具有相似色彩分布特征的目标内容图像。

9.一种计算机可读介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至7中任一项所述的方法。

10.一种电子设备，其特征在于，包括：

处理器；以及

存储器，用于存储所述处理器的可执行指令；

其中，所述处理器配置为经由执行所述可执行指令来执行权利要求1至7任一项所述的方法。