[发明专利]一种基于Transformer的图像颜色增强方法

摘要

本发明公开了一种基于Transformer的图像颜色增强方法。首先，使用相机拍摄自然场景的图像，并邀请专业的修图人员对图像的颜色进行修饰，构建带标注的图像数据集；然后，使用数据集内的图像训练一个基于Transformer的图像颜色增强深度模型，估计出图像颜色的转换矩阵；最后，将图像输入颜色增强模型内，得到图像的颜色转换矩阵，并以该矩阵与图像的多项式拓展结果间的乘积作为图像的颜色增强结果。本发明可利用Transformer的全局注意力机制来避免卷积模型的局部推断偏置所带来的约束，提升模型的表达力，进而训练出泛化能力更强的颜色增强模型。本发明在国际通用的数据集上，与其它方法进行了客观和主观比较，结果表明本方法可得到更为优越的颜色增强效果。