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

摘要

本申请公开了一种图像分割模型的训练方法及相关装置，方法包括对测试图像集中的测试图像进行数据增强得到若干增强测试图像；通过预分割模型确定测试图像的预测概率图及各增强测试图像的预测概率图，并基于各预测概率图确定母版概率图；基于母版概率图确定若干最大联合掩码图并基于若干最大联合掩码图确定伪标签；基于各伪标签微调分割模型的模型参数，以得到图像分割模型。本申请通过自我监督学习方来在线调节分割模型，并使用测试时期数据增强的方式来生成可靠的伪标签来驱动分割模型的在线学习，并根据伪标签的一致性程度来动态地停止学习，从而可以提高图像分割模型的模型性能以及泛化程度。