[发明专利]肤色分割方法、装置、电子设备及存储介质

说明书

本公开关于一种肤色分割方法、装置、电子设备及存储介质。其中方法通过获取待分割图像中每个像素的色度分量信息，并将像素的色度分量信息分别输入前景概率查找表和背景概率查找表进行查找，得到像素对应的前景概率和像素对应的背景概率，若像素对应的前景概率和像素对应的背景概率之间的差值满足设定条件，则标记像素为皮肤。由于本实施例是通过前景概率查找表和背景概率查找表对待分割图像中的每一个像素进行肤色识别分割，从而具有较高的准确率，能够实现精度更高的肤色分割。

技术领域

本公开涉及图像处理技术领域，尤其涉及一种肤色分割方法、装置、电子设备及存储介质。

背景技术

随着图像处理技术的发展，在用户进行拍照或是录制视频时，可以通过应用中提供的美颜功能对拍摄到的图像或图像序列进行美化处理，例如，祛除皮肤中比较明显的瑕疵，如痘、痣、斑等。

相关技术中，为了对图像或图像序列中的皮肤区域进行美化处理，首先需要对图像中的皮肤区域进行分割。传统的肤色分割算法如肤色椭圆模型认为，皮肤像素在YCbCr色彩模式中投影到Cb(蓝色色度分量)和Cr(红色色度分量)维度构成的二维平面中将被包含于一个椭圆中，该椭圆大小位置等参数由设计者的经验知识确定。这类方法虽然简洁，但在实际应用中准确度较低。因此，如何精确而高效的进行肤色分割是目前亟需解决的问题。

发明内容

本公开提供一种肤色分割方法、装置、电子设备及存储介质，以至少解决相关技术中肤色分割准确度低的问题。本公开的技术方案如下：

根据本公开实施例的第一方面，提供一种肤色分割方法，包括：

获取待分割图像中每个像素的色度分量信息；

将所述像素的色度分量信息分别输入前景概率查找表和背景概率查找表进行查找，得到所述像素对应的前景概率和所述像素对应的背景概率，所述前景概率查找表包括二维色彩空间中每一个像素点的色度分量信息以及所述像素点为肤色的概率值，所述背景概率查找表包括所述二维色彩空间中每一个像素点的色度分量信息以及所述像素点不为肤色的概率值，所述二维色彩空间由具有不同色度分量信息的像素点构成；

获取所述像素对应的前景概率和所述像素对应的背景概率之间的差值；

当所述差值大于或等于设定阈值时，标记所述像素为皮肤。

在其中一个实施例中，所述方法还包括：当所述差值小于所述设定阈值时，标记所述像素为非皮肤。

在其中一个实施例中，所述获取待分割图像中每个像素的色度分量信息，包括：获取待分割图像的色彩模式；当所述待分割图像的色彩模式为YCbCr色彩模式时，从所述待分割图像中提取每个像素的色度分量信息，所述色度分量信息包括蓝色色度分量Cb和红色色度分量Cr。

在其中一个实施例中，所述获取待分割图像的色彩模式之后，所述方法还包括：当所述待分割图像的色彩模式不是YCbCr色彩模式时，将所述待分割图像的色彩模式转换为所述YCbCr色彩模式；从转换为所述YCbCr色彩模式的待分割图像中提取每个像素的色度分量信息。

在其中一个实施例中，所述前景概率查找表的生成方法包括：获取第一样本数据集，所述第一样本数据集中包括若干个肤色样本像素，每个所述肤色样本像素具有对应的样本色度分量信息；根据所述第一样本数据集训练高斯混合模型，直到所述高斯混合模型收敛，得到目标前景高斯模型；将所述二维色彩空间中每一个像素点的色度分量信息输入所述目标前景高斯模型，得到所述二维色彩空间中每一个像素点为肤色的概率值；基于所述二维色彩空间中每一个像素点的色度分量信息以及所述像素点为肤色的概率值，生成对应的前景概率查找表。