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

说明书

本公开涉及一种图像分割方法及装置、电子设备和存储介质，所述方法包括：确定待分割图像中目标对象的人脸位置信息，所述目标对象是所述待分割图像中包括的多个对象中的至少一个；根据所述目标对象的人脸位置信息，对所述待分割图像进行分割，得到所述目标对象对应的分割结果。本公开实施例可以根据目标对象的人脸位置信息，从待分割图像中包括的多个对象中直接分割出目标对象，而无需对待分割图像中的多个对象都进行分割之后再对目标对象对应的分割结果进行筛选，从而可以对目标对象的分割效率，降低耗时。

技术领域

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

背景技术

图像分割是当前计算机视觉领域的一个重要应用，特别是对人像图片/视频内容进行后处理编辑时，把人像像素区域分割出来是最基本的一步。对于多人图片/视频，如何快速把需要的主体人物进一步区分出来是一个亟需解决的重要问题。

发明内容

本公开提出了一种图像分割方法及装置、电子设备和存储介质的技术方案。

根据本公开的一方面，提供了一种图像分割方法，包括：确定待分割图像中目标对象的人脸位置信息，所述目标对象是所述待分割图像中包括的多个对象中的至少一个；根据所述目标对象的人脸位置信息，对所述待分割图像进行分割，得到所述目标对象对应的分割结果。

通过确定目标对象的人脸位置信息，以及将人脸位置信息作为先验信息，使得可以根据目标对象的人脸位置信息，从待分割图像中包括的多个对象中直接分割出目标对象，而无需对待分割图像中的多个对象都进行分割之后再对目标对象对应的分割结果进行筛选，从而可以提高对目标对象的分割效率，降低耗时。

在一种可能的实现方式中，所述确定待分割图像中目标对象的人脸位置信息，包括：对所述待分割图像进行人脸检测，得到多个人脸框；响应于被选中的人脸框，将所述被选中的人脸框确定为所述目标对象对应的目标人脸框，所述目标人脸框用于指示所述待分割图像中所述目标对象的人脸位置信息。

通过对待分割图像进行人脸检测得到多个人脸框，以及响应于被选中的人脸框，可以快速将被选中的人脸框确定为，作为后续图像分割的人脸先验信息的目标对象对应的目标人脸框。

在一种可能的实现方式中，所述确定待分割图像中目标对象的人脸位置信息，包括：接收用户对所述目标对象的人脸区域的标注信息；根据所述标注信息，确定所述目标对象对应的目标人脸框，所述目标人脸框用于指示所述待分割图像中所述目标对象的人脸位置信息。

通过接收用户对目标对象的人脸区域的标注信息，从而可以根据标注信息，快速确定出作为后续图像分割的人脸先验信息的目标对象对应的目标人脸框。

在一种可能的实现方式中，所述根据所述目标对象的人脸位置信息，对所述待分割图像进行分割，得到所述目标对象对应的分割结果，包括：根据所述目标人脸框，生成所述目标人脸框对应的第一掩膜图像，所述第一掩膜图像中包括第一区域和第二区域，所述第一区域在所述第一掩膜图像中的位置与所述目标人脸框在所述待分割图像中的位置相同，所述第二区域为所述第一掩膜图像中所述第一区域以外的区域；基于所述第一掩膜图像，对所述待分割图像进行分割，得到所述目标对象对应的分割结果。

根据目标对象对应的目标人脸框生成第一掩膜图像之后，使得可以基于第一掩膜图像对待分割图像中的目标对象进行直接分割，得到目标对象对应的分割结果，而无需对待分割图像中的多个对象都进行分割之后再对目标对象对应的分割结果进行筛选，从而可以提高对目标对象的分割效率，降低耗时。

在一种可能的实现方式中，所述基于所述第一掩膜图像，对所述待分割图像进行分割，得到所述目标对象对应的分割结果，包括：将所述第一掩膜图像和所述待分割图像进行融合，得到融合后图像；基于所述融合后图像，利用训练好的深度神经网络对所述待分割图像进行分割，得到所述目标对象对应的分割结果。