【技术实现步骤摘要】
本专利技术涉及图像处理，尤其涉及一种局部图像条件着色方法、系统和电子设备。
技术介绍
1、
2、相关技术中，采用基于点提示的方法进行图像局部条件着色，希望依靠transfomer的良好的感受野来提升局部区域颜色的扩散能力，这种完全依赖网络能力去进行局部颜色扩散很难很好地落地，因为网络能力终究是有限的。另外，目前基于条件的着色方法主要通过点提示，即在目标区域手动选择几个点，希望颜色能够围绕着这些点向周围扩散，从而实现整个局部区域着色。因此，图像局部条件着色存在扩散不充分的问题。
技术实现思路
1、有鉴于此，为解决上述问题之一，本专利技术实施例的目的是提供一种局部图像条件着色方法、系统和电子设备，对局部目标着色区域实现精准着色。
2、一方面，本专利技术实施例提供了一种局部图像条件着色方法，包括：
3、根据局部区域对原图像进行处理，得到第一掩码；
4、根据预设颜色和第一掩码的提示点集合确定第二图像，根据所述提示点集合确定第一图像；
5、根据所述原图像、所述第一图像和所述第二图像确定第三图像。
6、在一个可选的实施例中，所述根据所述原图像、所述第一图像和所述第二图像确定第三图像，包括：
7、根据所述原图像和所述预设颜色确定变换系数；
8、根据所述原图像、所述第一图像、所述第二图像和所述变换系数确定目标着色区域着色后的第三图像。
9、在一个可选的实施例中，根据所述原图像、所述第一图像、
10、根据所述原图像的亮度图像、所述第一图像和所述第二图像确定包含色度信息的第四图像；
11、根据所述变换系数调整所述原图像，得到包含亮度信息的第五图像；
12、将所述第四图像和所述第五图像合并，确定目标着色区域着色后的第三图像。
13、在一个可选的实施例中，所述方法还包括：
14、将所述第一掩码、所述原图像和所述第三图像进行融合，得到着色后的第六图像。
15、在一个可选的实施例中，所述根据局部区域对原图像进行处理，得到第一掩码，包括：
16、根据所述局部区域对所述原图像进行分割，得到预设大小的局部图像；
17、将所述局部图像输入到语义分割模型，得到第一掩码。
18、在一个可选的实施例中，所述第一掩码的提示点集合通过以下方法确定，包括：
19、按照预设步长和预设顺序对所述第一掩码进行遍历；
20、将所述第一掩码中满足预设要求的像素点确定所述提示点集合。
21、在一个可选的实施例中，所述根据所述提示点集合确定第一图像，包括：
22、初始化一个像素值相同的初始图像；
23、将所述提示点集合在所述初始图像对应位置的像素值进行重置，确定所述第一图像。
24、在一个可选的实施例中，所述根据预设颜色和第一掩码的提示点集合确定第二图像，包括：
25、根据所述预设颜色构建第七图像，对所述第七图像进行通道分离，得到包含色度信息的第八图像；
26、将所述第一掩码的提示点集合在所述第八图像对应位置的颜色值保留，确定第二图像。
27、所述根据所述原图像和所述预设颜色确定变换系数，包括：
28、根据所述原图像的亮度均值和所述预设颜色的亮度均值进行第一运算，确定变换系数。
29、在一个可选的实施例中，所述将所述原图像的亮度图像、所述第一图像和所述第二图像确定包含色度信息的第四图像，包括：
30、将所述原图像的亮度图像、所述第一图像和所述第二图像进行拼接，得到第九图像；
31、将所述第九图像输入上色模型，确定包含色度信息的第四图像。
32、在一个可选的实施例中，所述根据所述变换系数调整所述原图像，得到包含亮度信息的第五图像，包括：
33、根据所述变换系数对所述原图像的亮度图像进行调整，得到包含亮度信息的第五图像。
34、在一个可选的实施例中，所述将所述第一掩码、所述原图像和所述第三图像进行融合，得到着色后的第六图像，包括：
35、根据所述第一掩码和所述第三图像确定目标着色区域内的第十图像；
36、根据所述第一掩码和所述原图像确定目标着色区域外的第十一图像；
37、将所述第十图像和所述第十一图像进行融合，得到着色后的第六图像。
38、另一方面，本专利技术实施例提供了一种局部图像条件着色系统，包括：
39、第一模块，用于根据局部区域对原图像进行处理，得到第一掩码；
40、第二模块，用于根据预设颜色和第一掩码的提示点集合确定第二图像，根据所述提示点集合确定第一图像；
41、第三模块，用于根据所述原图像、所述第一图像和所述第二图像确定第三图像。
42、在一个可选的实施例中，所述第三模块用于：
43、根据所述原图像和所述预设颜色确定变换系数；
44、根据所述原图像、所述第一图像、所述第二图像和所述变换系数确定目标着色区域着色后的第三图像。
45、在一个可选的实施例中，所述第三模块用于：
46、根据所述原图像的亮度图像、所述第一图像和所述第二图像确定包含色度信息的第四图像；
47、根据所述变换系数调整所述原图像，得到包含亮度信息的第五图像；
48、将所述第四图像和所述第五图像合并，确定目标着色区域着色后的第三图像。
49、在一个可选的实施例中，所述局部图像条件着色系统还包括第四模块，所述第四模块用于：
50、将所述第一掩码、所述原图像和所述第三图像进行融合，得到着色后的第六图像。
51、在一个可选的实施例中，所述第一模块用于：
52、根据所述局部区域对所述原图像进行分割，得到预设大小的局部图像；
53、将所述局部图像输入到语义分割模型，得到第一掩码。
54、在一个可选的实施例中，所述第二模块用于：
55、按照预设步长和预设顺序对所述第一掩码进行遍历；
56、将所述第一掩码中满足预设要求的像素点确定所述提示点集合。
57、在一个可选的实施例中，所述第二模块用于：
58、初始化一个像素值相同的初始图像；
59、将所述提示点集合在所述初始图像对应位置的像素值进行重置，确定所述第一图像。
60、在一个可选的实施例中，所述第二模块用于：
61、根据所述预设颜色构建第七图像，对所述第七图像进行通道分离，得到包含色度信息的第八图像；
62、将所述第一掩码的提示点集合在所述第八图像对应位置的颜色值保留，确定第二图像。
63、在一个可选的实施例中，所述第三模块用于：
64、根据所述原图像的亮度本文档来自技高网...
【技术保护点】
1.一种局部图像条件着色方法，其特征在于，包括：
2.根据权利要求1所述的方法，其特征在于，所述根据所述原图像、所述第一图像和所述第二图像确定第三图像，包括：
3.根据权利要求2所述的方法，其特征在于，根据所述原图像、所述第一图像、所述第二图像和所述变换系数确定目标着色区域着色后的第三图像，包括：
4.根据权利要求1-3任一项所述的方法，其特征在于，所述方法还包括：
5.根据权利要求1所述的方法，其特征在于，所述根据局部区域对原图像进行处理，得到第一掩码，包括：
6.根据权利要求1所述的方法，其特征在于，所述第一掩码的提示点集合通过以下方法确定：
7.根据权利要求1所述的方法，其特征在于，所述根据所述提示点集合确定第一图像，包括：
8.根据权利要求1所述的方法，其特征在于，所述根据预设颜色和第一掩码的提示点集合确定第二图像，包括：
9.根据权利要求2所述的方法，其特征在于，所述根据所述原图像和所述预设颜色确定变换系数，包括：
10.根据权利要求3所述的方法，其特征在于，所述
11.根据权利要求3所述的方法，其特征在于，所述根据所述变换系数调整所述原图像，得到包含亮度信息的第五图像，包括：
12.根据权利要求4所述的方法，其特征在于，所述将所述第一掩码、所述原图像和所述第三图像进行融合，得到着色后的第六图像，包括：
13.一种局部图像条件着色系统，其特征在于，包括：
14.一种电子设备，其特征在于，包括：
...【技术特征摘要】
1.一种局部图像条件着色方法，其特征在于，包括：
2.根据权利要求1所述的方法，其特征在于，所述根据所述原图像、所述第一图像和所述第二图像确定第三图像，包括：
3.根据权利要求2所述的方法，其特征在于，根据所述原图像、所述第一图像、所述第二图像和所述变换系数确定目标着色区域着色后的第三图像，包括：
4.根据权利要求1-3任一项所述的方法，其特征在于，所述方法还包括：
5.根据权利要求1所述的方法，其特征在于，所述根据局部区域对原图像进行处理，得到第一掩码，包括：
6.根据权利要求1所述的方法，其特征在于，所述第一掩码的提示点集合通过以下方法确定：
7.根据权利要求1所述的方法，其特征在于，所述根据所述提示点集合确定第一图像，包括：
8.根据...
【专利技术属性】
技术研发人员：张亮，刘长波，
申请(专利权)人：珠海金山办公软件有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。