本技术涉及图像处理领域,尤其涉及一种抠图方法、电子设备及计算机可读存储介质。
背景技术:
1、在图像处理领域,对原始图像进行抠图处理是指将目标区域的图像从原始图像中分离出来,从而可以将目标区域的图像用于图像合成或者对目标区域的图像进行编辑。现有的抠图方法,为了保证抠图精度,抠图算法的计算量较大,因此需要将抠图算法部署在云端,在端侧离线的情况下无法完成抠图。
技术实现思路
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、可以理解的是,上述第二方面至第六方面的有益效果可以参见上述第一方面中的相关描述,在此不再赘述。