[发明专利]自动抠图方法及系统

说明书

本发明公开了一种自动抠图方法、介质、设备及系统，其中方法包括：获取历史图片，并根据历史图片生成训练数据集；根据训练数据集进行分割模型的训练，以便通过分割模型生成对应的三元组图片；根据训练数据集中的原始图片和该原始图片对应的三元组图片进行抠图模型的训练；获取待抠图图片，并将待抠图图片输入到分割模型，以生成待抠图图片对应的三元组图片；将待抠图图片和该待抠图图片对应的三元组图片输入到抠图模型，以通过抠图模型生成该待抠图图片对应的图形蒙版，以及根据图形蒙版对待抠图图片进行自动抠图；能够根据用户输入的原始图片进行前景部分的自动抠取，降低用户抠图所需编辑时间，降低抠图难度；同时，保证抠图结果的稳定性。

技术领域

本发明涉及图像处理技术领域，特别涉及一种自动抠图方法、一种计算机可读存储介质、一种计算机设备以及一种自动抠图系统。

背景技术

抠图(图像去背景)是指准确提取静止图片或者视频图片序列中的前景目标，它是许多图像编辑中的关键技术。

相关技术中，在对静止图片或者视频图片进行抠图处理的过程中，多采用人工对静止图片或者视频图片中的前景部分进行抠取，而这一过程将耗费用户大量的编辑时间，并且操作门栏较高。同时，由于人为的不可确定性，将导致抠图结果不稳定，可能导致最终抠图结果不够精细。

发明内容

本发明旨在至少在一定程度上解决上述技术中的技术问题之一。为此，本发明的一个目的在于提出一种自动抠图方法，能够根据用户输入的原始图片进行前景部分的自动抠取，降低用户抠图所需编辑时间，降低抠图难度；同时，保证抠图结果的稳定性。

本发明的第二个目的在于提出一种计算机可读存储介质。

本发明的第三个目的在于提出一种计算机设备。

本发明的第四个目的在于提出一种自动抠图系统。

为达到上述目的，本发明第一方面实施例提出了一种自动抠图方法，包括以下步骤：获取历史图片，并根据所述历史图片生成训练数据集；根据所述训练数据集进行分割模型的训练，以便通过训练得到的分割模型根据训练数据集中的原始图片生成对应的三元组图片；根据所述训练数据集中的原始图片和该原始图片对应的三元组图片进行抠图模型的训练，以生成抠图模型；获取待抠图图片，并将所述待抠图图片输入到所述分割模型，以通过所述分割模型生成所述待抠图图片对应的三元组图片；将所述待抠图图片和该待抠图图片对应的三元组图片输入到所述抠图模型，以通过所述抠图模型生成该待抠图图片对应的图形蒙版，以及根据所述图形蒙版对所述待抠图图片进行自动抠图。

根据本发明实施例的自动抠图方法，首先，获取历史图片，并根据所述历史图片生成训练数据集；接着，根据所述训练数据集进行分割模型的训练，以便通过训练得到的分割模型根据训练数据集中的原始图片生成对应的三元组图片；然后，根据所述训练数据集中的原始图片和该原始图片对应的三元组图片进行抠图模型的训练，以生成抠图模型；接着，获取待抠图图片，并将所述待抠图图片输入到所述分割模型，以通过所述分割模型生成所述待抠图图片对应的三元组图片；然后，将所述待抠图图片和该待抠图图片对应的三元组图片输入到所述抠图模型，以通过所述抠图模型生成该待抠图图片对应的图形蒙版，以及根据所述图形蒙版对所述待抠图图片进行自动抠图；从而实现根据用户输入的原始图片进行前景部分的自动抠取，降低用户抠图所需编辑时间，降低抠图难度；同时，保证抠图结果的稳定性。

另外，根据本发明上述实施例提出的自动抠图方法还可以具有如下附加的技术特征：

可选地，根据所述历史图片生成训练数据集之前，还包括：计算每个所述历史图片对应的信噪比，并根据所述信噪比对所述历史图片进行过滤；对过滤后的历史图片中的显著性前景进行标注，以便根据标注后的历史图片生成训练数据集。

可选地，在对过滤后的历史图片中的显著性前景进行标注之后，还包括：获取背景数据集，并根据所述背景数据集对标注后的历史图片中的背景进行随机替换，以生成相应的扩充样本，以便根据标注后的历史图片和扩充样本生成训练数据集。