[发明专利]图像分割数据处理方法、装置、设备及存储介质

说明书

本申请实施例公开了一种图像分割数据处理方法、装置、计算机设备及存储介质，属于计算机技术领域。该方法包括：基于源域标签图像、第一标签图像和第二标签图像确定第一源域损失值；基于第三标签图像和第四标签图像确定第一目标域损失值；基于第一源域损失值和第一目标域损失值训练第一图像分割模型，并基于训练后得到的图像分割模型进行图像分割任务处理。采用源域图像和目标域图像对第一图像分割模型进行训练时，不仅考虑了该第一图像分割模型输出的结果，还将第二图像分割模型输出的结果融入到源域损失值中，能够通过第二图像分割模型对第一图像分割模型进行监督，训练出准确性和鲁棒性更高的图像分割模型，从而提高图像分割的准确性。

技术领域

本申请实施例涉及计算机技术领域，特别涉及一种图像分割数据处理方法、装置、设备及存储介质。

背景技术

随着计算机技术和人工智能技术的不断发展，人工智能技术在图像分割、图像识别等图像分析任务中取得了显著的成果。在采用人工智能技术进行图像分割时，需要依赖大量样本图像训练图像分割模型，但是样本图像中往往会存在噪声图像。

相关技术中，先基于质量评估模型确定样本图像的质量分数，选取质量分数较高的样本图像，利用质量分数较高的样本图像来训练图像分割模型。由于仅依赖于质量评估模型对样本图像进行筛选，筛选出来的样本图像的可靠性并不高，因此训练得到的图像分割模型的准确性和鲁棒性较差。

发明内容

本申请实施例提供了一种图像分割数据处理方法、装置、计算机设备及存储介质，能够图像分割模型的准确性和鲁棒性。所述技术方案如下：

一方面，提供了一种图像分割数据处理方法，所述方法包括：

获取第一源域图像、所述第一源域图像对应的源域标签图像、第一标签图像和第二标签图像，所述第一标签图像为基于第一图像分割模型对所述第一源域图像进行图像分割得到的，所述第二标签图像为基于第二图像分割模型对所述第一源域图像进行图像分割得到的；

基于所述第一源域图像对应的源域标签图像、所述第一标签图像和所述第二标签图像，确定第一源域损失值；

获取目标域图像、第三标签图像和第四标签图像，所述第三标签图像为基于所述第一图像分割模型对所述目标域图像进行图像分割得到的，所述第四标签图像为基于所述第二图像分割模型对所述目标域图像进行图像分割得到的；

基于所述第三标签图像和所述第四标签图像，确定第一目标域损失值；

基于所述第一源域损失值和所述第一目标域损失值，训练所述第一图像分割模型，并基于训练后得到的图像分割模型进行图像分割任务处理。

可选地，所述基于所述第二源域图像对应的源域标签图像和所述第五标签图像，确定第二源域损失值，包括：响应于所述相似度小于所述相似度阈值，基于所述第二源域图像对应的源域标签图像、所述第五标签图像和所述第六标签图像，确定所述第二源域损失值。

可选地，所述方法还包括：响应于所述迭代训练次数大于所述第二阈值，基于所述第三源域损失值和所述第四源域损失值，确定所述第一源域损失值。

可选地，所述基于所述源域标签图像，确定所述每个像素点对应的权重，包括：确定所述每个像素点与所述源域标签图像的边界线之间的最小距离；将确定的多个最小距离中的最大值确定为目标距离；基于所述目标距离和所述每个像素点对应的最小距离，分别确定所述每个像素点对应的权重。

可选地，所述基于所述每个像素点对应的第一标签和权重，确定所述第三源域损失值，包括：基于所述每个像素点对应的第一标签和权重，确定第一差异值，所述第一差异值表示所述第一标签图像和所述源域标签图像之间的差异；基于所述第一差异值、任一像素点对应的第一标签和权重，确定所述任一像素点对应的第三源域子损失值；基于确定的多个第三源域子损失值，确定所述第三源域损失值。