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

说明书

本发明提供了一种图像分割方法、装置及电子设备，涉及图像处理技术领域。本发明实施例所提供的图像分割方法、装置及电子设备，将待处理图像输入第一分割网络，得到所述待处理图像的初级分割图像；将所述初级分割图像与所述待处理图像进行级联，生成第一级联图像；将所述第一级联图像输入第二分割网络，得到所述待处理图像的精分割图像；获取所述待处理图像的边缘图像；所述边缘图像标示出所述待处理图像中目标对象区域与背景区域之间的过渡区域；根据所述精分割图像和所述边缘图像，获取所述待处理图像的分割结果。该方法可以使分割得到的目标对象区域边缘更加清晰和平滑，提高图像分割的精度。

技术领域

本发明涉及图像处理技术领域，尤其是涉及一种图像分割方法、装置及电子设备。

背景技术

图像分割是指将图像中的目标对象区域与背景区域进行分割。例如，将图像中的人物区域与背景区域分割，提取出图像中的人物与另一背景相结合，可以替换图像中的背景，生成新的图像。

现有的图像分割方法一般通过神经网络先将一幅图像进行编码，然后进行解码得到分割结果。通过该方法得到的目标对象区域的边缘不精确，因此图像分割的效果不佳。

发明内容

有鉴于此，本发明的目的在于提供一种图像分割方法、装置及电子设备，可以提高图像分割的精度。

为了实现上述目的，本发明实施例采用的技术方案如下：

第一方面，本发明实施例提供了一种图像分割方法，包括：

将待处理图像输入第一分割网络，得到所述待处理图像的初级分割图像；

将所述初级分割图像与所述待处理图像进行级联，生成第一级联图像；

将所述第一级联图像输入第二分割网络，得到所述待处理图像的精分割图像；

获取所述待处理图像的边缘图像；所述边缘图像标示出所述待处理图像中目标对象区域与背景区域之间的过渡区域；

根据所述精分割图像和所述边缘图像，获取所述待处理图像的分割结果。

结合第一方面，本发明实施例提供了第一方面的第一种可能的实施方式，其中，获取所述待处理图像的边缘图像的步骤，包括：

采用边缘检测算法确定所述待处理图像中目标对象区域的边界区域，得到所述待处理图像的边缘图像。

结合第一方面，本发明实施例提供了第一方面的第二种可能的实施方式，其中，获取所述待处理图像的边缘图像的步骤，包括：

将所述第一级联图像输入边缘网络，得到所述待处理图像的边缘图像。

结合第一方面，本发明实施例提供了第一方面的第三种可能的实施方式，其中，根据所述精分割图像和所述边缘图像，获取所述待处理图像的分割结果的步骤，包括：

将所述精分割图像与所述边缘图像进行级联，生成第二级联图像；

将所述第二级联图像作为所述待处理图像的分割结果。

结合第一方面或第一方面的第二种可能的实施方式，本发明实施例提供了第一方面的第四种可能的实施方式，其中，根据所述精分割图像和所述边缘图像，获取所述待处理图像的分割结果的步骤，包括：

将所述精分割图像与所述边缘图像进行级联，生成第二级联图像；

将所述第二级联图像输入第三分割网络，得到所述待处理图像的分割结果。

结合第一方面的第四种可能的实施方式，本发明实施例提供了第一方面的第五种可能的实施方式，其中，所述第一分割网络、第二分割网络、边缘网络或第三分割网络采用全卷积神经网络。