基于图像分割的图像抠图方法、装置、计算机设备及介质与流程

1.本技术涉及人工智能技术领域，尤其涉及一种基于图像分割的图像抠图方法、装置、计算机设备及介质。


背景技术：

2.图像抠图(image matting)是指对于一张给定的图像，网络可以分离其前景区域和背景区域，是目前计算机视觉领域的重要课题，广泛应用于视频会议、图像编辑和后期制作等场景中。目前，图像抠图技术通常使用额外输入，如：trimap、背景图像等，通过额外输入来实现蒙版的生成，利用蒙版提取抠图对象。
3.但是，额外输入的生成往往需要人工干预，例如trimap等先验蒙版的生成；有时额外输入的获取并不总是可行的，例如完整背景图像的获取，这使得图像抠图费时费力，同时，造成图像抠图效率低下以及抠图结果不准确。


技术实现要素：

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.图1是本技术可以应用于其中的示例性系统架构图；
50.图2是根据本技术的基于图像分割的图像抠图方法的一个实施例的流程图；
51.图3是图2中步骤s203的一种具体实施方式的流程图；
52.图4是根据本技术的基于图像分割的图像抠图方法的另一个实施例的流程图；
53.图5是根据本技术的基于图像分割的图像抠图装置的一个实施例的结构示意图；
54.图6是根据本技术的计算机设备的一个实施例的结构示意图。
具体实施方式
55.除非另有定义，本文所使用的所有的技术和科学术语与属于本技术的技术领域的技术人员通常理解的含义相同；本文中在申请的说明书中所使用的术语只是为了描述具体的实施例的目的，不是旨在于限制本技术；本技术的说明书和权利要求书及上述附图说明中的术语“包括”和“具有”以及它们的任何变形，意图在于覆盖不排他的包含。本技术的说明书和权利要求书或上述附图中的术语“第一”、“第二”等是用于区别不同对象，而不是用于描述特定顺序。
56.在本文中提及“实施例”意味着，结合实施例描述的特定特征、结构或特性可以包含在本技术的至少一个实施例中。在说明书中的各个位置出现该短语并不一定均是指相同的实施例，也不是与其它实施例互斥的独立的或备选的实施例。本领域技术人员显式地和隐式地理解的是，本文所描述的实施例可以与其它实施例相结合。
57.为了使本技术领域的人员更好地理解本技术方案，下面将结合附图，对本技术实施例中的技术方案进行清楚、完整地描述。
58.本技术提供了一种基于图像分割的图像抠图方法，涉及人工智能，可以应用于如图1所示的系统架构100中，系统架构100可以包括终端设备101、102、103，网络104和服务器105。网络104用以在终端设备101、102、103和服务器105之间提供通信链路的介质。网络104可以包括各种连接类型，例如有线、无线通信链路或者光纤电缆等等。
59.用户可以使用终端设备101、102、103通过网络104与服务器105交互，以接收或发送消息等。终端设备101、102、103上可以安装有各种通讯客户端应用，例如网页浏览器应用、购物类应用、搜索类应用、即时通信工具、邮箱客户端、社交平台软件等。
60.终端设备101、102、103可以是具有显示屏并且支持网页浏览的各种电子设备，包括但不限于智能手机、平板电脑、电子书阅读器、mp3播放器(moving picture experts group audio layer iii，动态影像专家压缩标准音频层面3)、mp4(moving picture experts group audio layer iv，动态影像专家压缩标准音频层面4)播放器、膝上型便携计算机和台式计算机等等。
61.服务器105可以是提供各种服务的服务器，例如对终端设备101、102、103上显示的页面提供支持的后台服务器。
62.需要说明的是，本技术实施例所提供的基于图像分割的图像抠图方法一般由服务器/终端设备执行，相应地，基于图像分割的图像抠图装置一般设置于服务器/终端设备中。
63.应该理解，图1中的终端设备、网络和服务器的数目仅仅是示意性的。根据实现需要，可以具有任意数目的终端设备、网络和服务器。
64.继续参考图2，示出了根据本技术的基于图像分割的图像抠图方法的一个实施例的流程图，包括以下步骤：
65.步骤s201，获取训练图像集，将训练图像集输入预构建的初始图像抠图模型，其中，图像抠图模型包括图像分割层和图像抠图层。
66.在本实施例中，预构建的初始图像抠图模型包括图像分割层和图像抠图层，其中，图像分割层可以采用double dip(deep-image-priors，深度图像先验)网络，double dip网络利用两个dip网络将输入图像分割为前景层和背景层；图像抠图层主干使用u-net网络进行编码解码，并在解码部分的后基层加入辅助输出层，进行深度监督，并加入渐进注意力细化模块(progressive attention refinement module，par)，利用解码器的中间层输出进行逐层细化，获得最终的精确蒙版，以得到精确的分割图像。
67.在本实施例中，训练图像集可以从公开数据集获取，例如，alphamatting数据集，alphamatting数据集包含27张训练图像和8张测试图像，这些图像均有抠图后前景和后景的标准结果图，然后，将这些图像的前景图分别与500张室内场景图像以及500张室外场景图像进行组合，对组合后的图像进行三个不同角度的旋转，这样得到的图像作为训练图像集和测试图像集；也可以根据获取的原始图片生成，具体的，获取原始图片(例如，人像图
片、商品图片、环境图片、动物图片、交通工具图片等)，计算每张原始图片对应的信噪比，并根据信噪比对原始图片进行过滤，对过滤后的原始图片中的显著性前景进行标注，以便根据标注后的原始图片生成训练数据集。
68.步骤s202，通过图像分割层对训练图像集中的图像进行分割，得到初步分割图像集。
69.在本实施例中，图像分割层将训练图像集中的训练图像进行前景层和背景层分割，并通过掩膜将前景层和背景层进行混合，得到重构图像，重构图像即为初步分割图像。
70.具体的，将训练图像集中的训练图像输入图像分割层，为每个图层分配一个dip网络，每个dip网络输入一个随机噪声zi，通过yi＝dip(zi)计算得到前景层y1和背景层y2，前景层和背景层两个图层通过掩膜m进行融合，得到重构图像i’，公式如下：
71.i'＝my1 (1-m)y272.应当理解，图像分割层为预训练完成的。
73.步骤s203，将初步分割图像集输入至图像抠图层，得到细化分割图像。
74.在本实施例中，图像抠图层至少包括编码器、解码器、渐进注意力细化层(progressive attention refinement，par)以及辅助输出层(output)，将初步分割图像输入图像抠图层进行细化分割，得到细化分割图像。
75.在一些可选的实现方式中，上述将初步分割图像集输入至图像抠图层，得到细化分割图像的步骤包括：
76.步骤s301，将初步分割图像集输入至编码器进行特征提取，得到编码特征。
77.其中，编码器包括多个卷积神经网络层(convolution neural network，cnn)和下采样层，下采样层可以为最大池化层(max-pooling)。通过cnn层提取空间特征，生成特征图，max-pooling层对特征图进行下采样，保留较强特征。
78.可选的，编码器包括5个卷积层和4个下采样层，5个卷积层分别为第一编码卷积层、第二编码卷积层、第三编码卷积层、第四编码卷积层和第五编码卷积层，第一编码卷积层与第二编码卷积层之间、第二编码卷积层与第三编码卷积层之间、第三编码卷积层与第四编码卷积层之间以及第四编码卷积层与第五编码卷积层之间，均设置有一个下采样层，分别为第一下采样层、第二下采样层、第三下采样层以及第四下采样层。
79.初步分割图像集中的初步分割图像依次通过第一编码卷积层、第一下采样层、第二编码卷积层、第二下采样层、第三编码卷积层、第三下采样层、第四编码卷积层、第四下采样层和第五编码卷积层进行特征提取，得到编码特征。
80.应当理解，编码器卷积层的卷积核和卷积步长可以根据实际情况进行设置。
81.步骤s302，通过解码器将编码特征进行解码，输出解码特征。
82.在本实施例中，解码器由多个解码模块组成，每个解码模块包括多个上采样层(up-sampling layer)和cnn层。每经过一次解码，特征图的大小增加相应倍数，经过多次解码，得到与原始输入的初步分割图像大小相同的特征图，即解码特征。此外，每次解码后的解码特征与编码阶段对应的同样大小的编码特征连接，以融合低级和高级特征。
83.可选的，解码器包括5个卷积层和4个上采样层，5个卷积层分别为第一解码卷积层、第二解码卷积层、第三解码卷积层、第四解码卷积层和第五解码卷积层，第一解码卷积层与第二解码卷积层之间、第二解码卷积层与第三解码卷积层之间、第三解码卷积层与第
四解码卷积层之间以及第四解码卷积层与第五解码卷积层之间，均设置有一个上采样层，分别为第一上采样层、第二上采样层、第三上采样层以及第四上采样层。
84.将编码特征输入解码器，依次通过第一解码卷积层、第一上采样层、第二解码卷积层、第二上采样层、第三解码卷积层、第三上采样层、第四解码卷积层、第四上采样层和第五解码卷积层进行解码，得到解码特征。
85.应当理解，解码器卷积层的卷积核和卷积步长也是可以根据实际情况进行设置。
86.步骤s303，将解码特征输入辅助输出层，得到输出特征。
87.在本实施例中，解码器每个卷积层后对应设置连接有一个辅助输出层，用于对输出的特征进行卷积池化操作，已保留图像更多的特征信息。
88.步骤s304，通过渐进注意力细化层对输出特征进行注意力计算，得到注意力特征，并根据注意力特征输出细化分割图像。
89.在本实施例中，渐进注意力层分别与前一层辅助输出层、与当前解码卷积层连接的当前解码辅助输出层以及渐进注意力层对应的辅助输出层(当前层辅助输出层)连接，具体的，渐进注意力层的输入为前一层辅助输出层以及当前解码辅助输出层的输出，进行注意力操作后，通过当前层辅助输出层输出。
90.需要说明的是，第一解码卷积层没有设置对应的渐进注意力层与其连接，如果第一解码卷积层为当前层，与其连接的辅助输出层既是当前解码输出层，也相当于当前层输出层。
91.渐进注意力层至少包括编码层、注意力卷积层、第一融合层、softmax层、第二融合层、连接层以及解码层，将输出特征依次通过编码层、注意力卷积层、第一融合层、softmax层、第二融合层、连接层以及解码层进行相应计算，输出更精确的细化分割图像。
92.步骤s204，基于细化分割图像确定目标损失函数，根据目标损失函数对初始图像抠图模型进行迭代更新，输出训练完成的图像抠图模型。
93.在本实施例中，将训练图像集输入至初始图像抠图模型中进行训练，一轮训练结束后，计算初始图像抠图模型的目标损失函数，得到损失函数值，根据损失函数值调整模型参数，继续进行迭代训练，模型训练到一定程度，此时，模型的性能达到最优状态，损失函数值无法继续下降，即收敛。而判断收敛的方式只需要计算前后两轮迭代中的损失函数值，若损失函数值仍在变化，则继续选择训练图像输入至调整模型参数后的图像抠图模型中，以对图像抠图模型继续进行迭代训练；若损失函数值没有显著变化，则可认为模型收敛。模型收敛后，输出最终的图像抠图模型。
94.步骤s205，获取目标图像，将目标图像输入图像抠图模型，得到抠图结果。
95.在本实施例中，将获取到的目标图像输入至训练完成的图像抠图模型中进行抠图操作，可以得到对应的抠图结果。
96.需要强调的是，为进一步保证目标图像的私密和安全性，上述目标图像还可以存储于一区块链的节点中。
97.本技术所指区块链是分布式数据存储、点对点传输、共识机制、加密算法等计算机技术的新型应用模式。区块链(blockchain)，本质上是一个去中心化的数据库，是一串使用密码学方法相关联产生的数据块，每一个数据块中包含了一批次网络交易的信息，用于验证其信息的有效性(防伪)和生成下一个区块。区块链可以包括区块链底层平台、平台产品
服务层以及应用服务层等。
98.本技术通过训练完成的图像抠图模型中的图像分割层对图像进行初步分割，再将初步分割结果输出图像抠图层进一步进行细化分割，可以实现无需任何额外输入来进行图像抠图，并且完全避免了人工干预，达到完全的自动化抠图，提高了图像抠图效率，同时，通过图像抠图模型可以达到更精确的抠图结果，进一步提高抠图精确度和准确率。
99.在本实施例的一些可选的实现方式中，在上述通过图像分割层对训练图像集中的图像进行分割，得到初步分割图像集的步骤之前还包括：
100.对图像分割层进行预训练，得到预训练完成的图像分割层，具体包括：
101.将获取到的原始图像集输入图像分割层，输出重构图像；
102.根据重构图像得到第一损失函数，基于第一损失函数对图像分割层进行迭代更新，输出预训练完成的图像分割层。
103.在本实施例中，原始图像集可以与上述训练数据集的获取渠道相同，也可以不同，根据实际需要进行选择。
104.将原始图像集中的原始图像输入图像分割层，分割为前景层y1和背景层y2，通过初始掩膜m0将前景层y1和背景层y2进行融合，得到重构图像i’。其中，初始掩膜m0可以是预先设置的，也可以是随机生成的，随机生成则是根据输入随机噪声获得。
105.在一些可选的实现方式中，上述根据重构图像得到第一损失函数，基于第一损失函数对图像分割层进行迭代更新，输出预训练完成的图像分割层的步骤包括：
106.根据重构图像和原始图像集中的原始图像计算得到重构损失；
107.根据重构损失确定第一损失函数；
108.基于第一损失函数调整图像分割层的分割参数；
109.当满足迭代结束条件时，根据分割参数输出预训练完成的图像分割层。
110.其中，第一损失函数采用如下公式计算：
111.loss
ddip
＝loss
reconst
β
·
loss
excl
γ
·
loss
reg
112.其中，loss
reconst
为重构损失，loss
reconst
＝||i-i'||，i为原始图像；loss
excl
为一种互斥损失，让y1和y2的梯度之间相关性最小化；loss
reg
＝(∑
x
|m(x)-0.5|)-1
，loss
reg
为正则化损失，主要是用来约束融合mask(掩膜)的，用于将前景初始掩膜m0二值化，β、γ为预设的加权参数。
113.计算出第一损失函数，基于第一损失函数调整图像分割层的分割参数，继续进行迭代训练，图像分割层训练到一定程度，此时，图像分割层的性能达到预设状态，预设状态可以是收敛，也可以是第一损失函数的损失函数值达到预设阈值，即说明满足迭代结束条件，图像分割层预训练完成，根据迭代结束时的分割参数输出图像分割层。
114.本实施例通过对图像分割层进行预训练，调整图像分割层的初始掩膜m0，预训练完成后，获得预训练的图像分割层的掩膜m，掩膜m在之后的训练过程中仍会进行学习调整，以便获得更加精确的前景掩膜m，用于后续图像抠图层。
115.在本实施例的一些可选的实现方式中，上述通过渐进注意力细化层对输出特征进行注意力计算，得到注意力特征的步骤包括：
116.步骤s401，将前一层辅助输出层输出的输出特征上采样到与当前层辅助输出层的输出特征相同尺寸，得到上采样输出特征，根据上采样输出特征计算得到不确定区域掩膜。
117.具体的，前一层辅助输出层输出的输出特征α
l-1
上采样到与当前辅助输出层输出特征α
l
相同尺寸，之后通过以下变换公式：
[0118][0119]
其中f
α
→m(x,y)为图像(x,y)点处由α蒙版获得不确定区域掩膜m的变换公式，α
l
为当前层辅助输出层l的α蒙版。通过可以获得前一层辅助输出层输出的不确定区域掩膜m
l-1
，以及当前层辅助输出层输出的不确定区域掩膜m
l
。
[0120]
步骤s402，对不确定区域掩膜进行特征提取，得到不确定区域特征，对不确定区域特征进行注意力计算，得到注意力分数。
[0121]
将前一层辅助输出层输出的不确定区域掩膜m
l-1
与当前层辅助输出层输出的不确定区域掩膜m
l
经过cnn组成的编码层分别进行特征提取，得到和然后通过如下公式，计算获得注意力分数，作为优化趋势作用并修正于当前层辅助输出层输出的不确定区域特征注意力分数计算公式如下：
[0122][0123]
其中，为1x1的卷积操作，为l-1层的不确定区域特征，为l层的不确定区域特征，x为计算的优化趋势注意力分数。
[0124]
步骤s403，根据注意力分数修正不确定区域特征，得到修正后的不确定区域特征作为注意力特征。
[0125]
采用如下公式对不确定区域特征进行修正：
[0126][0127]
其中，为1x1的卷积操作，为修正后的l层的不确定区域特征。
[0128]
本实施例通过注意力计算修正不确定区域特征得到注意力特征，可以保证后续细化抠图更加精确。
[0129]
在一些可选的实现方式中，上述根据注意力特征输出细化分割图像的步骤包括：
[0130]
对当前层辅助输出层的输出特征进行特征提取，得到提取特征，将注意力特征与提取特征进行拼接，得到拼接特征；
[0131]
将拼接特征进行解码，输出细化分割图像。
[0132]
具体的，将当前层辅助输出层输出α
l
通过由cnn组成的编码层进行特征提取，将提取的特征f
α
与获得的修正后的当前层辅助输出层注意力特征进行特征拼接，如下公式：
[0133][0134]
其中，为注意力特征，f
α
为当前层辅助输出层α
l
蒙版经过编码层提取的特征，f
α
'为修正的当前辅助输出层α蒙版特征，即拼接特征。
[0135]
通过由cnn组成的解码层对拼接特征f
α
′
进行解码，并获得细化分割图像。
[0136]
在本实施例的一些可选的实现方式中，上述基于细化分割图像确定目标损失函
数，根据目标损失函数对初始图像抠图模型进行迭代更新，输出训练完成的图像抠图模型的步骤包括：
[0137]
根据细化分割图像得到第二损失函数；
[0138]
基于第一损失函数和第二损失函数确定目标损失函数；
[0139]
根据调整初始图像抠图模型的模型参数；
[0140]
当满足迭代结束条件时，根据模型参数生成训练完成的图像抠图模型。
[0141]
具体的，第二损失函数的计算公式如下：
[0142][0143]
其中，ω
l
为超参；α
gt
表示当前层辅助输出层的输出真值；loss
l
表示当前层辅助输出层的不确定区域的损失，包括l1损失、组合损失和拉普拉斯损失，公式为：
[0144]
loss
l
(α
gt
·ml
,α
l
·ml
)＝loss
l1
(α
gt
·ml
,α
l
·ml
) loss
comp
(α
gt
·ml
,α
l
·ml
) loss
lap
(α
gt
·ml
,α
l
·ml
)
[0145]
基于第一损失函数和第二损失函数确定目标损失函数，则目标损失函数计算公式如下：
[0146]
loss＝δloss
ddip
ε
·
loss
matting
[0147]
其中，δ和ε为预设的加权参数。
[0148]
在本实施例中，根据损失函数值调整模型参数，继续进行迭代训练，模型训练到一定程度，此时，模型的性能达到最优状态，损失函数值无法继续下降，即收敛。
[0149]
满足迭代结束条件即为模型收敛，模型收敛后，根据最终调整的模型参数输出最终的图像抠图模型。
[0150]
本实施例通过训练预构建的图像抠图模型，可以提高图像抠图模型的抠图精确度和准确率。
[0151]
本技术实施例可以基于人工智能技术对相关的数据进行获取和处理。其中，人工智能(artificial intelligence，ai)是利用数字计算机或者数字计算机控制的机器模拟、延伸和扩展人的智能，感知环境、获取知识并使用知识获得最佳结果的理论、方法、技术及应用系统。
[0152]
人工智能基础技术一般包括如传感器、专用人工智能芯片、云计算、分布式存储、大数据处理技术、操作/交互系统、机电一体化等技术。人工智能软件技术主要包括计算机视觉技术、机器人技术、生物识别技术、语音处理技术、自然语言处理技术以及机器学习/深度学习等几大方向。
[0153]
本技术可用于众多通用或专用的计算机系统环境或配置中。例如：个人计算机、服务器计算机、手持设备或便携式设备、平板型设备、多处理器系统、基于微处理器的系统、置顶盒、可编程的消费电子设备、网络pc、小型计算机、大型计算机、包括以上任何系统或设备的分布式计算环境等等。本技术可以在由计算机执行的计算机可执行指令的一般上下文中描述，例如程序模块。一般地，程序模块包括执行特定任务或实现特定抽象数据类型的例程、程序、对象、组件、数据结构等等。也可以在分布式计算环境中实践本技术，在这些分布式计算环境中，由通过通信网络而被连接的远程处理设备来执行任务。在分布式计算环境
中，程序模块可以位于包括存储设备在内的本地和远程计算机存储介质中。
[0154]
本领域普通技术人员可以理解实现上述实施例方法中的全部或部分流程，是可以通过计算机可读指令来指令相关的硬件来完成，该计算机可读指令可存储于一计算机可读取存储介质中，该程序在执行时，可包括如上述各方法的实施例的流程。其中，前述的存储介质可为磁碟、光盘、只读存储记忆体(read-only memory，rom)等非易失性存储介质，或随机存储记忆体(random access memory，ram)等。
[0155]
应该理解的是，虽然附图的流程图中的各个步骤按照箭头的指示依次显示，但是这些步骤并不是必然按照箭头指示的顺序依次执行。除非本文中有明确的说明，这些步骤的执行并没有严格的顺序限制，其可以以其他的顺序执行。而且，附图的流程图中的至少一部分步骤可以包括多个子步骤或者多个阶段，这些子步骤或者阶段并不必然是在同一时刻执行完成，而是可以在不同的时刻执行，其执行顺序也不必然是依次进行，而是可以与其他步骤或者其他步骤的子步骤或者阶段的至少一部分轮流或者交替地执行。
[0156]
进一步参考图5，作为对上述图2所示方法的实现，本技术提供了一种基于图像分割的图像抠图装置的一个实施例，该装置实施例与图2所示的方法实施例相对应，该装置具体可以应用于各种电子设备中。
[0157]
如图5所示，本实施例所述的基于图像分割的图像抠图装置500包括：获取模块501、初步分割模块502、细化分割模块503、训练模块504以及抠图模块505。其中：
[0158]
获取模块501用于获取训练图像集，将所述训练图像集输入预构建的初始图像抠图模型，其中，图像抠图模型包括图像分割层和图像抠图层。
[0159]
初步分割模块502用于通过所述图像分割层对所述训练图像集中的图像进行分割，得到初步分割图像集；
[0160]
细化分割模块503用于将所述初步分割图像集输入至所述图像抠图层，得到细化分割图像；
[0161]
训练模块504用于基于所述细化分割图像确定目标损失函数，根据所述目标损失函数对所述初始图像抠图模型进行迭代更新，输出训练完成的图像抠图模型；
[0162]
抠图模块505用于获取目标图像，将所述目标图像输入图像抠图模型，得到抠图结果。
[0163]
上述基于图像分割的图像抠图装置，通过训练完成的图像抠图模型中的图像分割层对图像进行初步分割，再将初步分割结果输出图像抠图层进一步进行细化分割，可以实现无需任何额外输入来进行图像抠图，并且完全避免了人工干预，达到完全的自动化抠图，提高了图像抠图效率，同时，通过图像抠图模型可以达到更精确的抠图结果，提高抠图精确度和准确率。
[0164]
在本实施例的一些可选的实现方式中，初步分割模块502包括重构子模块和预训练子模块，其中，重构子模块用于将获取到的原始图像集输入图像分割层，输出重构图像；预训练子模块用于根据所述重构图像得到第一损失函数，基于所述第一损失函数对所述图像分割层进行迭代更新，输出预训练完成的所述图像分割层。
[0165]
本实施例通过对图像分割层进行预训练，调整图像分割层的初始掩膜，预训练完成后，获得预训练的图像分割层的掩膜，保证在之后的训练过程中获得更加精确的前景掩膜，用于后续图像抠图层进行抠图。
[0166]
在本实施例中，细化分割模块503包括特征提取子模块、解码子模块、输出子模块以及注意力子模块，其中：
[0167]
特征提取子模块用于将所述初步分割图像集输入至所述编码器进行特征提取，得到编码特征；
[0168]
解码子模块用于通过所述解码器将所述编码特征进行解码，输出解码特征；
[0169]
输出子模块用于将所述解码特征输入所述辅助输出层，得到输出特征；
[0170]
注意力子模块用于通过所述渐进注意力细化层对所述输出特征进行注意力计算，得到注意力特征，并根据所述注意力特征输出所述细化分割图像。
[0171]
本实施例通过注意力计算得到注意力特征，可以输出更精确的细化分割图像。
[0172]
在本实施例中，注意力子模块包括上采样单元、注意力计算单元以及修正单元，其中：
[0173]
上采样单元用于将前一层辅助输出层输出的输出特征上采样到与当前层辅助输出层的输出特征相同尺寸，得到上采样输出特征，根据所述上采样输出特征计算得到不确定区域掩膜；
[0174]
注意力计算单元用于对所述不确定区域掩膜进行特征提取，得到不确定区域特征，对所述不确定区域特征进行注意力计算，得到注意力分数；
[0175]
修正单元用于根据所述注意力分数修正不确定区域特征，得到修正后的不确定区域特征作为注意力特征。
[0176]
本实施例通过注意力计算修正不确定区域特征得到注意力特征，可以保证后续细化抠图更加精确。
[0177]
在一些可选的实现方式中，注意力子模块还包括拼接单元和解码单元，其中：
[0178]
拼接单元用于对当前层辅助输出层的输出特征进行特征提取，得到提取特征，将所述注意力特征与所述提取特征进行拼接，得到拼接特征；
[0179]
解码单元用于将所述拼接特征进行解码，输出所述细化分割图像。
[0180]
在本实施例中，训练模块504包括损失函数计算子模块、调整子模块以及生成子模块，其中：
[0181]
损失函数计算子模块用于根据所述细化分割图像得到第二损失函数；
[0182]
损失函数计算子模块还用于基于所述第一损失函数和所述第二损失函数确定目标损失函数；
[0183]
调整子模块用于根据所述调整所述初始图像抠图模型的模型参数；
[0184]
生成子模块用于当满足迭代结束条件时，根据所述模型参数生成训练完成的图像抠图模型。
[0185]
本实施例通过训练预构建的图像抠图模型，可以提高图像抠图模型的抠图精确度和准确率。
[0186]
在本实施例中，预训练子模块包括计算单元、调整单元以及输出单元，其中：
[0187]
计算单元用于根据所述重构图像和所述原始图像集中的原始图像计算得到重构损失；
[0188]
计算单元还用于根据所述重构损失确定所述第一损失函数；
[0189]
调整单元用于基于所述第一损失函数调整所述图像分割层的分割参数；
[0190]
输出单元用于当满足迭代结束条件时，根据所述分割参数输出预训练完成的图像分割层。
[0191]
本实施例通过根据重构损失确定的第一损失函数进行参数调整，可以提升预训练效率，同时保证后续训练获得更精确的掩膜。
[0192]
为解决上述技术问题，本技术实施例还提供计算机设备。具体请参阅图6，图6为本实施例计算机设备基本结构框图。
[0193]
所述计算机设备6包括通过系统总线相互通信连接存储器61、处理器62、网络接口63。需要指出的是，图中仅示出了具有组件61-63的计算机设备6，但是应理解的是，并不要求实施所有示出的组件，可以替代的实施更多或者更少的组件。其中，本技术领域技术人员可以理解，这里的计算机设备是一种能够按照事先设定或存储的指令，自动进行数值计算和/或信息处理的设备，其硬件包括但不限于微处理器、专用集成电路(application specific integrated circuit，asic)、可编程门阵列(field－programmable gate array，fpga)、数字处理器(digital signal processor，dsp)、嵌入式设备等。
[0194]
所述计算机设备可以是桌上型计算机、笔记本、掌上电脑及云端服务器等计算设备。所述计算机设备可以与用户通过键盘、鼠标、遥控器、触摸板或声控设备等方式进行人机交互。
[0195]
所述存储器61至少包括一种类型的可读存储介质，所述可读存储介质包括闪存、硬盘、多媒体卡、卡型存储器(例如，sd或dx存储器等)、随机访问存储器(ram)、静态随机访问存储器(sram)、只读存储器(rom)、电可擦除可编程只读存储器(eeprom)、可编程只读存储器(prom)、磁性存储器、磁盘、光盘等。在一些实施例中，所述存储器61可以是所述计算机设备6的内部存储单元，例如该计算机设备6的硬盘或内存。在另一些实施例中，所述存储器61也可以是所述计算机设备6的外部存储设备，例如该计算机设备6上配备的插接式硬盘，智能存储卡(smart media card,smc)，安全数字(secure digital,sd)卡，闪存卡(flash card)等。当然，所述存储器61还可以既包括所述计算机设备6的内部存储单元也包括其外部存储设备。本实施例中，所述存储器61通常用于存储安装于所述计算机设备6的操作系统和各类应用软件，例如基于图像分割的图像抠图方法的计算机可读指令等。此外，所述存储器61还可以用于暂时地存储已经输出或者将要输出的各类数据。
[0196]
所述处理器62在一些实施例中可以是中央处理器(central processing unit，cpu)、控制器、微控制器、微处理器、或其他数据处理芯片。该处理器62通常用于控制所述计算机设备6的总体操作。本实施例中，所述处理器62用于运行所述存储器61中存储的计算机可读指令或者处理数据，例如运行所述基于图像分割的图像抠图方法的计算机可读指令。
[0197]
所述网络接口63可包括无线网络接口或有线网络接口，该网络接口63通常用于在所述计算机设备6与其他电子设备之间建立通信连接。
[0198]
本实施例通过处理器执行存储在存储器的计算机可读指令时实现如上述实施例基于图像分割的图像抠图方法的步骤，通过训练完成的图像抠图模型中的图像分割层对图像进行初步分割，再将初步分割结果输出图像抠图层进一步进行细化分割，可以实现无需任何额外输入来进行图像抠图，并且完全避免了人工干预，达到完全的自动化抠图，提高了图像抠图效率，同时，通过图像抠图模型可以达到更精确的抠图结果，提高抠图精确度和准确率。
[0199]
本技术还提供了另一种实施方式，即提供一种计算机可读存储介质，所述计算机可读存储介质存储有计算机可读指令，所述计算机可读指令可被至少一个处理器执行，以使所述至少一个处理器执行如上述的基于图像分割的图像抠图方法的步骤，通过训练完成的图像抠图模型中的图像分割层对图像进行初步分割，再将初步分割结果输出图像抠图层进一步进行细化分割，可以实现无需任何额外输入来进行图像抠图，并且完全避免了人工干预，达到完全的自动化抠图，提高了图像抠图效率，同时，通过图像抠图模型可以达到更精确的抠图结果，提高抠图精确度和准确率。
[0200]
通过以上的实施方式的描述，本领域的技术人员可以清楚地了解到上述实施例方法可借助软件加必需的通用硬件平台的方式来实现，当然也可以通过硬件，但很多情况下前者是更佳的实施方式。基于这样的理解，本技术的技术方案本质上或者说对现有技术做出贡献的部分可以以软件产品的形式体现出来，该计算机软件产品存储在一个存储介质(如rom/ram、磁碟、光盘)中，包括若干指令用以使得一台终端设备(可以是手机，计算机，服务器，空调器，或者网络设备等)执行本技术各个实施例所述的方法。
[0201]
显然，以上所描述的实施例仅仅是本技术一部分实施例，而不是全部的实施例，附图中给出了本技术的较佳实施例，但并不限制本技术的专利范围。本技术可以以许多不同的形式来实现，相反地，提供这些实施例的目的是使对本技术的公开内容的理解更加透彻全面。尽管参照前述实施例对本技术进行了详细的说明，对于本领域的技术人员来而言，其依然可以对前述各具体实施方式所记载的技术方案进行修改，或者对其中部分技术特征进行等效替换。凡是利用本技术说明书及附图内容所做的等效结构，直接或间接运用在其他相关的技术领域，均同理在本技术专利保护范围之内。