[发明专利]人像抠图方法和装置

说明书

本申请实施例公开了人像抠图方法和装置。该方法的一具体实施方式包括：获取呈现有人像的原始图像；将原始图像输入到预先训练的三分图生成模型中，得到原始图像的三分图；基于原始图像、原始图像的三分图和预先训练的抠图模型，确定原始图像的蒙版；利用原始图像的蒙版，从原始图像中截取出人像图像。该实施方式可以使得三分图生成模型和抠图模型没有耦合，方便对任一模型进行修改或增强功能。

技术领域

本申请实施例涉及计算机技术领域，具体涉及人像抠图方法和装置。

背景技术

随着智能手机以及互联网技术的迅速发展，用手机拍照并对照片进行编辑分享已经成为人们生活中的重要组成部分。人像照片作为照片的一种形式，在日常娱乐、社交以及分享中扮演着重要角色。人像照片一般是指人像加背景的形式的照片。对于这样的图片，有一种比较常见的需求是将人像和背景分离，即我们常说的人像分割或人像抠图。

发明内容

本申请实施例提出了人像抠图方法和装置。

第一方面，本申请实施例提供了一种人像抠图方法，包括：获取呈现有人像的原始图像；将原始图像输入到预先训练的三分图生成模型中，得到原始图像的三分图；基于原始图像、原始图像的三分图和预先训练的抠图模型，确定原始图像的蒙版；利用原始图像的蒙版，从原始图像中截取出人像图像。

在一些实施例中，基于原始图像、原始图像的三分图和预先训练的抠图模型，确定原始图像的蒙版，包括：对原始图像的三分图进行修正；将原始图像和修正后的三分图输入到预先训练的抠图模型中，得到原始图像的蒙版。

在一些实施例中，对原始图像的三分图进行修正，包括：确定原始图像的三分图中前景区域的至少一个连通区域；将前景区域的至少一个连通区域中区域面积小于预设的第一面积阈值的连通区域确定为第一目标连通区域；将第一目标连通区域由前景区域更改为未知区域，得到修正后的三分图。

在一些实施例中，对原始图像的三分图进行修正，包括：确定原始图像的三分图中背景区域的至少一个连通区域；将背景区域的至少一个连通区域中区域面积小于预设的第二面积阈值的连通区域确定为第二目标连通区域；将第二目标连通区域由背景区域更改为未知区域，得到修正后的三分图。

在一些实施例中，三分图生成模型是通过如下方式训练得到的：获取第一训练样本集合，其中，第一训练样本包括第一样本图像和第一样本三分图，第一样本三分图是通过对第一样本图像对应的蒙版进行形态学变换所生成的；将第一训练样本集合中的第一训练样本中的第一样本图像和第一样本三分图分别作为第一初始模型的输入和期望输出，利用机器学习方法训练第一初始模型，得到三分图生成模型。

在一些实施例中，抠图模型是通过如下方式训练得到的：获取第二训练样本集合，其中，第二训练样本包括第二样本图像、第二样本三分图和样本蒙版，其中，第二样本图像中呈现有人像；将第二训练样本集合中的第二训练样本中的第二样本图像和第二样本三分图作为第二初始模型的输入，将与输入的第二样本图像和第二样本三分图对应的样本蒙版作为第二初始模型的期望输出，利用机器学习方法训练第二初始模型，得到抠图模型。

第二方面，本申请实施例提供了一种人像抠图装置，包括：获取单元，被配置成获取呈现有人像的原始图像；输入单元，被配置成将原始图像输入到预先训练的三分图生成模型中，得到原始图像的三分图；确定单元，被配置成基于原始图像、原始图像的三分图和预先训练的抠图模型，确定原始图像的蒙版；截取单元，被配置成利用原始图像的蒙版，从原始图像中截取出人像图像。

在一些实施例中，确定单元进一步被配置成通过如下方式基于原始图像、原始图像的三分图和预先训练的抠图模型，确定原始图像的蒙版：对原始图像的三分图进行修正；将原始图像和修正后的三分图输入到预先训练的抠图模型中，得到原始图像的蒙版。