[发明专利]目标主体分割方法、装置、设备及存储介质

权利要求书

1.一种目标主体分割方法，其特征在于，包括：

获取待分割图像，所述待分割图像中包括背景图像和目标主体图像；

对所述待分割图像进行目标主体特征提取，以得到目标主体分割图像，所述目标主体分割图像中像素的像素值为所述待分割图像中对应像素位于所述目标主体图像内的概率；

对所述目标主体分割图像进行后处理，以滤除所述目标主体分割图像中的干扰像素，所述干扰像素为所述目标主体分割图像中对应于所述背景图像的像素。

2.根据权利要求1所述的方法，其特征在于，所述对所述目标主体分割图像进行后处理，包括：

对所述目标主体分割图像进行阈值分割处理，以得到二值化的阈值分割图像。

3.根据权利要求2所述的方法，其特征在于，在所述对所述目标主体分割图像进行阈值分割处理，以得到二值化的阈值分割图像之后，所述方法还包括：

对所述阈值分割图像进行腐蚀操作和膨胀操作。

4.根据权利要求1至3任一项所述的方法，其特征在于，所述对所述待分割图像进行目标主体特征提取，以得到目标主体分割图像，包括：

对所述待分割图像进行多层级目标主体特征提取，以分别得到多个目标主体特征图像；

对多个所述目标主体特征图像中最深层级的目标主体特征图像进行金字塔池化处理；

对多个所述目标主体特征图像以及金字塔池化处理后的目标主体特征图像进行特征融合，以得到所述目标主体分割图像。

5.一种目标主体分割装置，其特征在于，包括：

获取模块，用于获取待分割图像，所述待分割图像中包括背景图像和目标主体图像；

特征提取模块，用于对所述待分割图像进行目标主体特征提取，以得到目标主体分割图像，所述目标主体分割图像中像素的像素值为所述待分割图像中对应像素位于所述目标主体图像内的概率；

后处理模块，用于对所述目标主体分割图像进行后处理，以滤除所述目标主体分割图像中的干扰像素，所述干扰像素为所述目标主体分割图像中对应于所述背景图像的像素。

6.根据权利要求5所述的装置，其特征在于，所述后处理模块，具体用于对所述目标主体分割图像进行阈值分割处理，以得到二值化的阈值分割图像。

7.根据权利要求6所述的装置，其特征在于，所述后处理模块，还具体用于对所述阈值分割图像进行腐蚀操作和膨胀操作。

8.根据权利要求5至7任一项所述的装置，其特征在于，所述特征提取模块包括编码模块、上下文模块以及解码模块；

所述编码模块，用于对所述待分割图像进行多层级目标主体特征提取，以分别得到多个目标主体特征图像；

所述上下文模块，用于对多个所述目标主体特征图像中最深层级的目标主体特征图像进行金字塔池化处理；

所述解码模块，用于对多个所述目标主体特征图像以及金字塔池化处理后的目标主体特征图像进行特征融合，以得到所述目标主体分割图像。

9.一种电子设备，包括：

至少一个处理器；以及

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

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行权利要求1-4中任一项所述的方法。

10.一种存储有计算机指令的非瞬时计算机可读存储介质，其中，所述计算机指令用于使所述计算机执行根据权利要求1-4中任一项所述的方法。

11.一种计算机程序产品，包括计算机程序，所述计算机程序在被处理器执行时实现根据权利要求1-4中任一项所述的方法。