[发明专利]图像分割方法及装置

权利要求书

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

响应于用户针对图像区域做出的点击操作，确定表征点击位置的第一编码图；

分别从原始图像、第一编码图和已有掩膜图中截取对应第一区域的部分，并将其输入第一语义分割模型，得到第一预测掩膜，以及所述第一区域的语义表征；其中所述第一区域是所述图像区域中包含已有掩膜的局部区域；

通过比对所述第一预测掩膜和已有掩膜，确定包含差异掩膜的第二区域；

分别从所述原始图像、第一编码图、第一预测掩膜和语义表征中截取对应第二区域的部分，并将其输入第二语义分割模型，得到第二预测掩膜，用于更新已有掩膜。

2.根据权利要求1所述的方法，其中，所述第一编码图包括针对前景和背景的两张编码图；其中，确定表征点击位置的第一编码图，包括：

将所述图像区域中对应点击位置和其他位置的像素块编码为不同码值，得到与所述点击操作所针对的前景或背景相对应的编码图；并且，将所述图像区域中的像素块编码为同一码值，得到另一编码图。

3.根据权利要求1所述的方法，其中，在分别从原始图像、第一编码图和已有掩膜图中截取对应第一区域的部分之前，所述方法还包括：

确定所述图像区域中包含所述已有掩膜且具有预设形状的最小面积区域；

对所述最小面积区域进行预设倍数的放大，作为所述第一区域。

4.根据权利要求1所述的方法，其中，在分别从原始图像、第一编码图和已有掩膜图中截取对应第一区域的部分之前，所述方法还包括：

在所述已有掩膜为无的情况下，将所述图像区域确定为所述第一区域。

5.根据权利要求1所述的方法，其中，在将其输入第一语义模型之前，所述方法还包括：

从表征所述图像区域的历史点击位置的编码矩阵中截取对应所述第一区域的部分。

6.根据权利要求1所述的方法，其中，所述第一语义分割模型包括表征层和预测层；其中，将其输入第一语义分割模型，得到第一预测掩膜，以及所述第一区域的语义表征，包括：

利用所述表征层处理所述第一语义分割模型的输入，得到所述语义表征；

利用所述预测层处理所述语义表征，得到所述第一预测掩膜。

7.根据权利要求1所述的方法，其中，通过比对所述第一预测掩膜和已有掩膜，确定包含差异掩膜的第二区域，包括：

通过比对所述第一预测掩膜和已有掩膜，确定差异掩膜，其包含多个分散掩膜；

确定所述多个分散掩膜中与所述点击位置相交且面积最大的目标分散掩膜；

基于所述目标分散掩膜，确定所述第二区域。

8.根据权利要求7所述的方法，其中，基于所述目标分散掩膜，确定所述第二区域，包括：

确定所述第一区域中包含所述目标分散掩膜且具有预设形状的最小面积区域；

对所述最小面积区域进行预设倍数的放大，作为所述第二区域。

9.根据权利要求1所述的方法，其中，所述语义表征包含对应多个通道的多个表征矩阵；其中，从所述语义表征中截取对应第二区域的部分，包括：

按照表征矩阵相较所述原始图像的缩小比例，对所述第二区域进行缩小处理，得到缩小后区域；

分别从所述多个表征矩阵中截取对应所述缩小后区域的部分，得到多个子矩阵；

根据预设维数，分别对所述多个子矩阵进行池化处理或插值处理，得到所述语义表征对应所述第二区域的部分。