基于参考图颜色风格的灰度图像着色方法技术

本发明专利技术公开了一种基于参考图颜色风格的灰度图像着色方法，可以作为线稿及黑白图像上色这一类问题的解决方案。本发明专利技术具体实施包括如下步骤：1.拟定上色图像类别库，根据类别收集彩色图像数据集，并对其进行线稿提取或灰度化，以及提取其颜色直方图信息。2.将彩色图像、线稿或灰度图像、颜色直方图信息作为训练集，输入网络中训练，得到对应该类别的着色模型。3.将待上色图像重新输入网络，使用步骤2中得到的模型进行图像着色。4.评估着色结果。本发明专利技术所提出的基于参考图颜色风格的灰度图像着色方法可以解决各种类别的线稿、黑白图像着色问题，并且在实践中证明着色效果良好。并且在实践中证明着色效果良好。并且在实践中证明着色效果良好。

【技术实现步骤摘要】
基于参考图颜色风格的灰度图像着色方法


[0001]本专利技术属于计算机技术及图像处理领域，尤其涉及一种基于参考图颜色风格的灰度图像着色方法。

技术介绍

[0002]图像上色是为了还原、增强原图像的表达效果，提高其视觉感受，还能丰富图像的语义，提高其应用价值。最早在19世纪，就流行以手工上色的方法来还原黑白照片等，随着技术发展，图像上色在影视处理、动漫艺术等领域得到了广泛的应用。如漫画行业，利用图像上色技术，能够解决大量黑白线稿上色费时费力等问题，节约绘制成本，且不影响购买者的阅读体验；在电影行业，早期黑白电影的观看体验较差，利用图像上色将黑白电影转为彩色电影，观众能在欣赏经典的同时获得视觉享受；对医学影像进行着色，也有助于医生对病人的病变部位有一个更加直观的观察和分析。
[0003]图像上色技术是图像处理领域一个活跃且具有挑战性的课题，传统着色方法分为两种，一类是基于颜色扩展的着色方法，另一类是基于颜色传递的着色方法。而机器学习的发展与在该课题上的应用，例如使用卷积神经网络学习从灰度图像到彩色图像的映射等，减轻了传统方法下人为工作量，并本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.基于参考图颜色风格的灰度图像着色方法，其特征在于，包括以下步骤：步骤1：构建特定类别的彩色图像数据库并进行预处理，采用双线性插值法调整图像大小至统一尺寸；步骤2：对步骤1中得到的彩色图像数据库进行线稿提取或图像灰度化，获得线稿图像或灰度图像数据集，并提取彩色图像的颜色直方图信息；所述的线稿提取采用XDoG算子，计算公式如下：d(i,j)＝img1(i,j)
‑
γ*img2(i,j)其中，(i,j)为像素点坐标；d为初始线稿图像；img1和img2为将彩色图像经过两个不同尺度的高斯差分滤波后得到图像，γ是用于调整高斯差分滤波的截止效果强度的参数；通过设定阈值ε，对提取的初始线稿图像进行阈值操作，得到最终线稿图像：其中，d
′
为经过阈值操作后的最终线稿图像，Ψ为用于平衡线稿图像亮度的软阈值参数；步骤3：构建条件生成对抗网络，包括生成器和鉴别器；将步骤2中得到的线稿图像与颜色直方图信息输入到生成器中进行卷积处理，然后将真实彩色图像、生成器生成的彩色图像、以及标签图像一起作为鉴别器的输入，对彩色图像的真假进行鉴别，完成对条件生成对抗网络的训练，将训练结束后的生成器作为对应类别图像的着色模型；所述的标签图像为步骤2中得到的线稿图像或灰度图像；步骤4：使用步骤3得到的着色模型对同类别的线稿图像或灰度图像进行着色，生成着色后的彩色图像。2.如权利要求1所述的基于参考图颜色风格的灰度图像着色方法，其特征在于，所述步骤3中的生成器为一个包含下采样和上采样的U
‑
Net结构，首先将输入m*n*3统一大小的线稿图像和颜色直方图信息经过1个卷积层和N个下采样层进行下采样操作；其中m和n分别为图像的长和宽，N为生成器网络结构的下采样层数；然后再采用N个上采样层进行上采样操作，并在前两层上采样层之后分别引入一层自注意力层；最后一层上采样层的输出结果再经过Relu激活层、卷积层、Tanh激活层最终生成处理后的彩色图像。3.如权利要求2所述的基于参考图颜色风格的灰度图像着色方法，其特征在于，所述的下采样和上采样层各取6层，输入图像大小为512*512*3。4.如权利要求1所述的基于参考图颜色风格的灰度图像着色方法，其特征在于，所述步骤3中的鉴别器对输入的图片进行分块预测，...

【专利技术属性】
技术研发人员：丁勇，阮翊婷，夏靖雯，汤峻，
申请(专利权)人：浙江大学，
类型：发明
国别省市：