对象分割方法和装置、电子设备及存储介质

专利主权项内容

微信公众号马克 数据网 1.一种对象分割方法，其特征在于，包括：获取待处理图像，所述待处理图像包含目标对象；将所述待处理图像输入到经训练的第一分割模型中，以获得所述目标对象的掩膜图像；其中，所述第一分割模型至少通过将所述目标对象的样本掩膜图像和对应的掩膜标签输入至成对相似性损失函数进行训练，所述样本掩膜图像是通过将包含所述目标对象的样本图像输入至待训练的所述第一分割模型而获得，训练所述第一分割模型时，使对于相同样本图像，所述成对相似性损失函数采用的掩膜标签通过第一掩膜特征图和第一边缘特征图进行学习，所述第一掩膜特征图和所述第一边缘特征图是通过将所述样本图像输入至第二分割模型而获得的；经训练的第一分割模型通过以下训练操作训练得到：获取所述样本图像；将所述样本图像输入预训练的所述第二分割模型中，以获得所述第二分割模型输出的所述目标对象的所述第一掩膜特征图和所述第一边缘特征图；将所述样本图像输入所述第一分割模型中，以获得所述目标对象的所述样本掩膜图像；基于所述成对相似性损失函数采用的掩膜标签获得第二掩膜特征图和第二边缘特征图；将所述第一掩膜特征图和所述第二掩膜特征图代入第一损失函数计算第一损失值；将所述第一边缘特征图和所述第二边缘特征图代入第一损失函数计算第二损失值；基于所述第一损失值和所述第二损失值对所述掩膜标签进行学习；将学习后的掩膜标签以及所述样本掩膜图像代入所述成对相似性损失函数进行计算，获得第三损失值；基于所述第三损失值对待训练的所述第一分割模型进行训练，获得经训练的所述第一分割模型。