[发明专利]图像分割方法、训练图像分割模型的方法及装置

权利要求书

1.一种图像分割方法，其特征在于，所述方法包括：

获取待分割图像作为查询图像，以及获取所述查询图像的支持图像以及查询图像对应的目标描述文本，其中所述支持图像与所述查询图像均包含所述目标描述文本所指示的目标；

利用所述目标描述文本针对所述查询图像和所述支持图像分别生成第一伪掩码，所述第一伪掩码为相应图像中目标区域被掩码后得到的；

对所述查询图像和所述支持图像分别进行编码处理，得到所述查询图像的第一特征表示和所述支持图像的第一特征表示；

利用所述查询图像的第一伪掩码和所述支持图像的第一伪掩码，对所述查询图像的第一特征表示和所述支持图像的第一特征表示进行相关性匹配；

利用所述相关性匹配的结果进行解码处理，得到对所述查询图像的分割结果，所述分割结果包括目标区域的位置信息。

2.根据权利要求1所述的方法，其特征在于，所述生成第一伪掩码包括：将所述查询图像和所述支持图像分别作为待掩码图像执行以下步骤：

将所述目标描述文本和待掩码图像输入掩码生成器，得到所述待掩码图像的第二伪掩码；

获取所述待掩码图像的第一特征表示；

利用所述第二伪掩码和所述待掩码图像的第一特征表示，获取所述待掩码图像的第一目标原型表示和第一背景原型表示；

利用所述待掩码图像的第一特征表示分别与所述第一目标原型表示、第一背景原型表示之间的相似度，得到所述待掩码图像的第一伪掩码。

3.根据权利要求2所述的方法，其特征在于，利用所述第二伪掩码和所述待掩码图像的第一特征表示，获取所述待掩码图像的第一目标原型表示包括：

利用所述查询图像的第二伪掩码和第一特征表示，得到所述查询图像的第二目标原型表示；以及利用所述支持图像的第二伪掩码和第一特征表示，得到所述支持图像的第二目标原型表示；将所述查询图像的第二目标原型表示和所述支持图像的第二目标原型表示进行加权处理，将加权处理后得到的表示作为所述待掩码图像的第一目标原型表示；

利用所述第二伪掩码和所述待掩码图像的第一特征表示，获取所述待掩码图像的第一背景原型表示包括：

利用所述待掩码图像的第一特征表示和第二伪掩码，得到所述待掩码图像的背景特征表示；利用第一映射值与所述待掩码图像的背景特征表示，得到所述待掩码图像的第一背景原型表示，其中所述第一映射值是利用所述待掩码图像的背景特征表示和第一特征表示进行映射后得到的。

4.根据权利要求1所述的方法，其特征在于，利用所述查询图像的第一伪掩码和所述支持图像的第一伪掩码，对所述查询图像的第一特征表示和所述支持图像的第一特征表示进行相关性匹配包括：

利用所述查询图像的第一伪掩码对所述查询图像的第一特征表示进行增强处理，以及利用所述支持图像的第一伪掩码对所述支持图像的第一特征表示进行增强处理；

利用增强后的所述查询图像的第一特征表示和增强后的所述支持图像的第一特征表示进行相关性匹配。

5.根据权利要求4所述的方法，其特征在于，所述利用增强后的所述查询图像的第一特征表示和增强后的所述支持图像的第一特征表示进行相关性匹配包括：

利用增强后的所述查询图像的第一特征表示和增强后的所述支持图像的第一特征表示进行交叉注意力处理；

利用所述交叉注意力处理的结果和所述支持图像的第一特征表示，得到所述支持图像的第三目标原型表示；

将所述支持图像的第三原型表示与增强后的所述查询图像的第一特征表示进行相关性匹配，和/或，将所述支持图像的第三原型表示与所述查询图像的第一特征表示进行相关性匹配。

6.根据权利要求1所述的方法，其特征在于，所述方法还包括：利用所述支持图像的第一伪掩码以及所述支持图像的第一特征表示，得到所述支持图像的N个第四目标原型表示，所述N为正整数；利用所述支持图像的N个第四目标原型表示和所述支持图像的第一特征表示的相关性，确定所述支持图像的目标关联表示；

利用所述相关性匹配的结果进行解码处理包括：利用所述相关性匹配的结果以及所述支持图像的目标关联表示进行解码处理。