[发明专利]图像分割模型的训练方法及相关设备

权利要求书

1.一种图像分割模型的训练方法，其特征在于，包括：

获取待分割图像和交互式表征图像；

基于随机选择方式在前景分割模式和交互式分割模式中确定当前分割模式，所述前景分割模式用于使用所述待分割图像对图像分割模型进行阶段训练，所述交互式分割模式用于使用所述待分割图像和所述交互式表征图像对所述图像分割模型进行阶段训练；

使用所述当前分割模式对所述图像分割模型进行阶段训练；

在所述阶段训练完成时，基于所述随机选择方式更新所述当前分割模式，以使用更新后的当前分割模式对所述图像分割模型进行训练，直至满足训练结束条件。

2.根据权利要求1所述的图像分割模型的训练方法，其特征在于，在所述当前分割模式为所述前景分割模式时，所述使用所述当前分割模式对所述图像分割模型进行阶段训练，包括：

获取预设掩膜图像、预设正交互表征图像和预设负交互表征图像，其中，所述预设掩膜图像、所述预设正交互表征图像和所述预设负交互表征图像的像素值均为指定像素值；

将所述待分割图像、所述预设掩膜图像、所述预设正交互表征图像和所述预设负交互表征图像进行合并，获得第一输入图像；

将所述第一输入图像输入至所述图像分割模型中，对所述图像分割模型进行阶段训练。

3.根据权利要求2所述的图像分割模型的训练方法，其特征在于，所述图像分割模型包括编码层和解码层；

所述将所述第一输入图像输入至所述图像分割模型中，对所述图像分割模型进行阶段训练，包括：

将所述第一输入图像输入至所述编码层进行编码处理，获得第一特征向量；

将所述第一特征向量输入至所述解码层进行解码处理，获得第一预测掩膜图像；

根据所述第一预测掩膜图像对所述图像分割模型进行阶段训练。

4.根据权利要求1或2所述的图像分割模型的训练方法，其特征在于，在所述当前分割模式为所述交互式分割模式时，所述交互式表征图像包括第一掩膜图像、第一正交互表征图像和第一负交互表征图像；所述使用所述当前分割模式对所述图像分割模型进行阶段训练，包括：

将所述待分割图像、所述第一掩膜图像、所述第一正交互表征图像和所述第一负交互表征图像进行合并，获得第二输入图像；

将所述第二输入图像输入至所述图像分割模型中，对所述图像分割模型进行阶段训练。

5.根据权利要求4所述的图像分割模型的训练方法，其特征在于，所述交互式表征图像还包括第二正交互表征图像和第二负交互表征图像；

所述将所述第二输入图像输入至所述图像分割模型中，对所述图像分割模型进行阶段训练，包括：

将所述第二输入图像输入至所述图像分割模型中，获得第二掩膜图像；

将所述待分割图像、所述第二掩膜图像、所述第二正交互表征图像和所述第二负交互表征图像进行合并，获得第三输入图像；

将所述第三输入图像输入至所述图像分割模型中，对所述图像分割模型进行阶段训练。

6.根据权利要求4所述的图像分割模型的训练方法，其特征在于，所述图像分割模型包括编码层和解码层；

所述将所述第二输入图像输入至所述图像分割模型中，对所述图像分割模型进行阶段训练，包括：

将所述第二输入图像输入至所述编码层进行编码处理，获得第二特征向量；

将所述第二特征向量输入至所述解码层进行解码处理，获得第二预测掩膜图像；

根据所述第二预测掩膜图像对所述图像分割模型进行阶段训练。

7.根据权利要求1所述的图像分割模型的训练方法，其特征在于，所述随机选择方式为以预设概率在所述前景分割模式和所述交互式分割模式中进行随机选择。