一种基于色彩恒常性的图像和谐化合成方法

摘要

本发明公开了一种基于色彩恒常性的图像和谐化合成方法，包括：将基于深度学习的粗调整网络模型和引入注意力机制的精细调整网络模型组成整体对抗生成网络，训练数据为包含成对真实图像和合成图像的公开图像数据集；以待调整的合成图像为原输入图像，获取灰度信息输入到粗调整网络模型中，输出图像光源色度的影响权重并据此估算光源颜色信息，对前景区域进行基于色彩平衡的调整得到粗糙和谐化的合成结果；利用精细调整网络模型，将粗调整的合成图像和对应掩码图通过转换相似的域特征得到色彩和谐的合成结果；通过高斯泊松方法结合精细调整的合成图像的颜色低频特征和原输入图像的梯度信息，共同优化生成和谐化合成图像。