[发明专利]人体掩膜提取方法及装置

说明书

本发明涉及计算机视觉技术领域，具体涉及一种人体掩膜提取方法及装置，该人体掩膜提取方法包括获取灰度图像和深度图像，将所述灰度图像输入预设的深度学习模型中，使用矩形框圈出包含人体图像的灰度图像。计算所述深度图像中包含人体图像的深度值范围，根据所述矩形框对所述灰度图像和深度图像进行裁剪，得到包含人体图像的灰度图像和包含人体图像的深度图像。对所述裁剪出的包含人体图像的灰度图像进行边缘检测，得到边缘图。根据所述边缘图的边缘点在所述裁剪得到的包含人体图像的深度图像中对应的深度值以及所述深度值范围，在所述裁剪得到的包含人体图像的灰度图像中确定一轮廓，选取所述轮廓上的像素点围成的最大连通区域为人体掩膜。

技术领域

本发明涉及计算机视觉技术领域，具体而言，涉及一种人体掩膜提取方法及装置。

背景技术

人体掩膜的提取是一种利用算法提取人体轮廓区域的技术，提取掩膜是对人体图像处理和分析的前提。现有的人体掩膜提取算法多是基于可见光图像的，这些提取算法主要有一下两类：一是先通过分割算法得到图像中的物体，然后将分割得到的物体进行识别得到人体掩膜；二是提取局部特征并通过分类算法进行训练得到模型，通过模型提取人体掩膜。第一类提取方法中的图像分割算法，因可见光图像易受光照和颜色、背景等因素的影响，分割出的图像会出现边界断点、图形合并等现象影响识别。而第二类提取方法中，因为人体形变和背景复杂，会导致准确率降低。因此，提供一种能精确提取人体掩膜的方法是十分必要的，能便于人体图像处理和分析。

发明内容

本发明的目的在于提供一种人体掩膜提取方法，以实现对人体掩膜的精确提取，以便于人体图像的处理和分析。

本发明的另一目的在于提供一种人体掩膜提取装置，以实现对人体掩膜的精确提取，以便于人体图像的处理和分析。

为了实现上述目的，本发明实施例采用的技术方案如下：

第一方面，本发明实施例提供了一种人体掩膜提取方法，应用于服务终端，所述方法包括：

获取灰度图像和深度图像；

将所述灰度图像输入预设的深度学习模型中，使用矩形框圈出包含人体图像的灰度图像；

根据所述矩形框计算所述深度图像中人体图像的深度值范围；

根据所述矩形框对所述灰度图像和所述深度图像进行裁剪，得到包含人体图像的灰度图像和包含人体图像的深度图像；

对所述裁剪出的包含人体图像的灰度图像进行边缘检测，得到边缘图；

根据所述边缘图的边缘点在所述裁剪得到的包含人体图像的深度图像中对应的深度值以及所述深度值范围，在所述裁剪得到的包含人体图像的灰度图像中确定一轮廓；

选取所述轮廓上的像素点围成的最大连通区域为人体掩膜。

第二方面，本发明实施例还提供了一种人体掩膜提取装置，应用于服务终端，所述装置包括：

图像获取模块，用于获取灰度图像和深度图像；

图像圈出模块，用于将所述灰度图像输入预设的深度学习模型中，使用矩形框圈出包含人体图像的灰度图像；

深度范围计算模块，用于根据所述矩形框计算所述深度图像中人体图像的深度值范围；

图像裁剪模块，用于根据所述矩形框对所述灰度图像和所述深度图像进行裁剪，得到包含人体图像的灰度图像和包含人体图像的深度图像；

边缘检测模块，用于对所述裁剪出的包含人体图像的灰度图像进行边缘检测，得到边缘图；

轮廓确定模块，用于根据所述边缘图的边缘点在所述裁剪得到的包含人体图像的深度图像中对应的深度值以及所述深度值范围，在所述裁剪得到的包含人体图像的灰度图像中确定一轮廓；