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

权利要求书

1.一种图像分割方法，包括：

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

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

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

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

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

2.如权利要求1所述的方法，其中，所述依据所述局部分割概率图对所述局部图像进行分割，得到第二分割概率图，包括：

生成局部分割概率图的三分图；

将所述局部图像和所述三分图作为第一分割网络的输入进行分割，得到第二分割概率图。

3.如权利要求1所述的方法，其中，所述将所述第一分割概率图和所述第二分割概率图进行结合，得到所述输入人物图像的分割结果，包括：

将所述第一分割概率图中的所述区域的图像内容替换为所述第二分割概率图的图像内容，得到所述输入人物图像的分割结果；或者

将所述第一分割概率图中的所述区域的第一图像内容替换为所述第二分割概率图的第一图像内容，将所述第一分割概率图中的所述区域的第二图像内容与所述第二分割概率图的第二图像内容进行融合，得到所述输入人物图像的分割结果，其中，所述第一分割概率图的第一图像内容在所述区域的位置与所述第二分割概率图的第一图像内容在所述第二分割概率图中的位置相同，所述第二分割概率图的第一图像内容在所述区域的位置与所述第二分割概率图的第二图像内容在所述第二分割概率图中的位置相同。

4.如权利要求1所述的方法，其中，所述第一分割概率图为所述输入人物图像中各像素点为人体部位的概率图，所述目标部位为头部，所述第一分割概率图为所述局部图像中各像素点为头部的概率图。

5.一种图像分割装置，包括：

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

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

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

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

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

6.如权利要求5所述的装置，其中，所述分割模块，包括：

生成单元，用于生成局部分割概率图的三分图；

分割单元，用于将所述局部图像和所述三分图作为第一分割网络的输入进行分割，得到第二分割概率图。

7.如权利要求5所述的装置，其中，所述结合模块用于将所述第一分割概率图中的所述区域的图像内容替换为所述第二分割概率图的图像内容，得到所述输入人物图像的分割结果；或者

所述结合模块用于将所述第一分割概率图中的所述区域的第一图像内容替换为所述第二分割概率图的第一图像内容，将所述第一分割概率图中的所述区域的第二图像内容与所述第二分割概率图的第二图像内容进行融合，得到所述输入人物图像的分割结果，其中，所述第一分割概率图的第一图像内容在所述区域的位置与所述第二分割概率图的第一图像内容在所述第二分割概率图中的位置相同，所述第二分割概率图的第一图像内容在所述区域的位置与所述第二分割概率图的第二图像内容在所述第二分割概率图中的位置相同。

8.如权利要求5所述的装置，其中，所述第一分割概率图为所述输入人物图像中各像素点为人体部位的概率图，所述目标部位为头部，所述第一分割概率图为所述局部图像中各像素点为头部的概率图。