[发明专利]一种人像提取方法、人像提取装置及移动终端

说明书

本申请公开了一种人像提取方法，人像提取装置、移动终端及计算机可读存储介质。该方法包括：获取当前帧的预览图像，其中，所述预览图像中包含有人像；对所述预览图像进行预处理，得到处理后图像；将所述处理后图像输入至人像分割模型，得到已分割图像，所述已分割图像包括背景区域、人体掩膜及头发掩膜；对所述头发掩膜作滤波处理，得到头发抠图结果；对所述头发抠图结果及所述已分割图像进行图像融合，得到目标图像。本申请方案基于局部抠图实现对头发区域的精细化处理，缓解了因单一人像分割技术所引起的头发区域的边缘瑕疵问题。

技术领域

本申请属于图像处理技术领域，尤其涉及一种人像提取方法、人像提取装置、移动终端及计算机可读存储介质。

背景技术

由于头发区域没有传统的边缘概念，因而，在使用传统的人像分割算法提取人像时，人像的头发区域往往会出现“过边缘”或“欠边缘”的情况。也即，当前的人像分割技术容易引起头发区域的边缘瑕疵问题。

发明内容

本申请提供了一种人像提取方法、人像提取装置、移动终端及计算机可读存储介质，可更加精准的从图像中提取出人像，减少人像的头发区域的边缘瑕疵问题。

第一方面，本申请提供了一种人像提取方法，包括

获取当前帧的预览图像，其中，上述预览图像中包含有人像；

对上述预览图像进行预处理，得到处理后图像；

将上述处理后图像输入至人像分割模型，得到已分割图像，上述已分割图像包括背景区域、人体掩膜及头发掩膜；

对上述头发掩膜作滤波处理，得到头发抠图结果；

对上述头发抠图结果及上述已分割图像进行图像融合，得到目标图像。

第二方面，本申请提供了一种人像提取装置，包括：

图像获取单元，用于获取当前帧的预览图像，其中，上述预览图像中包含有人像；

图像预处理单元，用于对上述预览图像进行预处理，得到处理后图像；

图像分割单元，用于将上述处理后图像输入至人像分割模型，得到已分割图像，上述已分割图像包括背景区域、人体掩膜及头发掩膜；

头发抠图单元，用于对上述头发掩膜作滤波处理，得到头发抠图结果；

图像融合单元，对上述头发抠图结果及上述已分割图像进行图像融合，得到目标图像。

第三方面，本申请提供了一种移动终端，上述移动终端包括存储器、处理器以及存储在上述存储器中并可在上述处理器上运行的计算机程序，上述处理器执行上述计算机程序时实现如上述第一方面的方法的步骤。

第四方面，本申请提供了一种计算机可读存储介质，上述计算机可读存储介质存储有计算机程序，上述计算机程序被处理器执行时实现如上述第一方面的方法的步骤。

第五方面，本申请提供了一种计算机程序产品，上述计算机程序产品包括计算机程序，上述计算机程序被一个或多个处理器执行时实现如上述第一方面的方法的步骤。

由上可见，通过本申请方案，首先获取当前帧的预览图像，其中，上述预览图像中包含有人像，然后对上述预览图像进行预处理，得到处理后图像，再将上述处理后图像输入至人像分割模型，得到已分割图像，上述已分割图像包括背景区域、人体掩膜及头发掩膜，接着对上述头发掩膜作滤波处理，得到头发抠图结果，最后对上述头发抠图结果及上述已分割图像进行图像融合，得到目标图像。通过本申请方案，在对图像进行了人像分割得到已分割图像后，再基于头发部分进行局部的抠图处理，并将抠图处理所得到头发抠图结果与已分割图像进行图像融合，以缓解因单一人像分割技术所引起的头发区域的边缘瑕疵问题。可以理解的是，上述第二方面至第五方面的有益效果可以参见上述第一方面中的相关描述，在此不再赘述。