[发明专利]图像分割方法、装置、电子设备和存储介质

说明书

本申请公开了图像分割方法、装置、电子设备和存储介质，涉及计算机视觉和深度学习技术等人工智能领域。具体实现方案为：获取输入人物图像的第一分割概率图；检测所述输入人物图像的目标部位所在区域，并获取所述区域对应的包括所述目标部位的局部图像；获取所述第一分割概率图中所述区域的局部分割概率图；依据所述局部分割概率图对所述局部图像进行分割，得到第二分割概率图；将第一分割概率图和第二分割概率图进行结合，得到所述输入人物图像的分割结果。本公开可以提高图像分割的精确度。

技术领域

本公开涉及计算机技术领域，尤其涉及计算机视觉和深度学习技术等人工智能领域。

背景技术

随着图像处理技术的发展，图像分割应用越来越广泛，例如：人像抠图。然而，目前主要图像分割技术是人工采用图像处理软件进行图像分割，或者传统的图像分割方法，例如：近邻算法(KNN)抠图技术。

发明内容

本公开提供了一种图像分割方法、装置、电子设备和存储介质。

根据本公开的一方面，提供了一种图像分割方法，包括：

获取输入人物图像的第一分割概率图；

检测所述输入人物图像的目标部位所在区域，并获取所述区域对应的包括所述目标部位的局部图像；

获取所述第一分割概率图中所述区域的局部分割概率图；

依据所述局部分割概率图对所述局部图像进行分割，得到第二分割概率图；

将第一分割概率图和第二分割概率图进行结合，得到所述输入人物图像的分割结果。

根据本公开的另一方面，提供了一种图像分割装置，包括：

第一获取模块，用于获取输入人物图像的第一分割概率图；

第二获取模块，用于检测所述输入人物图像的目标部位所在区域，并获取所述区域对应的包括所述目标部位的局部图像；

第三获取模块，用于获取所述第一分割概率图中所述区域的局部分割概率图；

分割模块，用于依据所述局部分割概率图对所述局部图像进行分割，得到第二分割概率图；

结合模块，用于将第一分割概率图和第二分割概率图进行结合，得到所述输入人物图像的分割结果。

根据本公开的另一方面，提供了一种电子设备，包括：

至少一个处理器；以及

与所述至少一个处理器通信连接的存储器；其中，

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行本公开提供的图像分割方法。

根据本公开的另一方面，提供了一种存储有计算机指令的非瞬时计算机可读存储介质，其特征在于，所述计算机指令用于使所述计算机执行本公开提供的图像分割方法。

根据本公开的技术方案，由于将第一分割概率图和第二分割概率图进行结合，得到输入人物图像的分割结果，从而可以提高图像分割的精确度。

应当理解，本部分所描述的内容并非旨在标识本公开的实施例的关键或重要特征，也不用于限制本公开的范围。本公开的其它特征将通过以下的说明书而变得容易理解。

附图说明

附图用于更好地理解本方案，不构成对本公开的限定。其中：

图1是本公开提供的一种图像分割方法的流程图；

图2是本公开提供的一种图像分割的示意图；

图3是本公开提供的一种图像分割装置的结构图；