[发明专利]人像提取方法、装置、电子设备及存储介质

权利要求书

1.一种人像提取方法，包括：

获取待处理的图像；

获取所述图像的语义分割结果和实例分割结果，其中，所述语义分割结果包括：所述图像中人像区域的掩膜图像；所述实例分割结果包括：所述图像中至少一个人像的掩膜图像；

对所述至少一个人像的掩膜图像以及所述人像区域的掩膜图像进行融合处理，以生成所述至少一个人像的融合后掩膜图像；

结合所述至少一个人像的融合后掩膜图像，提取所述图像中的所述至少一个人像。

2.根据权利要求1所述的人像提取方法，其中，所述对所述至少一个人像的掩膜图像以及所述人像区域的掩膜图像进行融合处理，以生成所述至少一个人像的融合后掩膜图像，包括：

根据所述至少一个人像的掩膜图像，确定所述至少一个人像的边缘框；

针对每个人像，获取所述人像的边缘框与所述图像中其它人像的边缘框之间的交集区域以及非交集区域；

获取所述人像的掩膜图像中位于所述交集区域的第一掩膜部分图像；

获取所述人像区域的掩膜图像中位于所述非交集区域的第二掩膜部分图像；

对所述第一掩膜部分图像和所述第二掩膜部分图像进行融合处理，以生成所述人像的融合后掩膜图像。

3.根据权利要求1所述的人像提取方法，其中，在获取所述图像的语义分割结果和实例分割结果之前，还包括：

按照预设尺寸对所述图像进行缩放处理，以得到具有所述预设尺寸的所述图像。

4.根据权利要求1或3所述的人像提取方法，其中，所述获取所述图像的语义分割结果和实例分割结果，包括：

将所述图像输入语义分割模型，以获取所述图像的语义分割结果；

将所述图像输入实例分割模型，以获取所述图像的实例分割结果。

5.根据权利要求1所述的人像提取方法，其中，在结合所述至少一个人像的融合后掩膜图像，提取所述图像中的所述至少一个人像之后，还包括：

获取所述图像对应的去遮挡背景图；

确定所述至少一个人像的目标位置；

根据所述去遮挡背景图、所述至少一个人像以及对应的目标位置，生成人像位置移动后的图像。

6.根据权利要求1所述的人像提取方法，其中，在结合所述至少一个人像的融合后掩膜图像，提取所述图像中的所述至少一个人像之后，还包括：

获取所述图像对应的去遮挡背景图；

确定所述至少一个人像中待添加到所述去遮挡背景图中的第一人像，以及所述第一人像的目标位置；

结合所述去遮挡背景图、所述第一人像以及对应的目标位置，生成具有所述第一人像的图像。

7.一种人像提取装置，包括：

获取模块，用于获取待处理的图像；

所述获取模块，还用于获取所述图像的语义分割结果和实例分割结果，其中，所述语义分割结果包括：所述图像中人像区域的掩膜图像；所述实例分割结果包括：所述图像中至少一个人像的掩膜图像；

融合处理模块，用于对所述至少一个人像的掩膜图像以及所述人像区域的掩膜图像进行融合处理，以生成所述至少一个人像的融合后掩膜图像；

提取模块，用于结合所述至少一个人像的融合后掩膜图像，提取所述图像中的所述至少一个人像。

8.根据权利要求7所述的人像提取装置，其中，所述获取模块具体用于，

根据所述至少一个人像的掩膜图像，确定所述至少一个人像的边缘框；

针对每个人像，获取所述人像的边缘框与所述图像中其它人像的边缘框之间的交集区域以及非交集区域；

获取所述人像的掩膜图像中位于所述交集区域的第一掩膜部分图像；

获取所述人像区域的掩膜图像中位于所述非交集区域的第二掩膜部分图像；

对所述第一掩膜部分图像和所述第二掩膜部分图像进行融合处理，以生成所述人像的融合后掩膜图像。

9.根据权利要求7所述的人像提取装置，其中，所述装置还包括：缩放处理模块，用于按照预设尺寸对所述图像进行缩放处理，以得到具有所述预设尺寸的所述图像。