模型训练方法、装置、计算机设备及计算机可读存储介质

本申请涉及一种模型训练方法、装置、计算机设备及计算机可读存储介质。方法应用于包括多个注意力模块的图像分割模型，每个注意力模块对应一张提示图像和一张提示标注掩膜，方法包括：获取训练图像和训练图像对应的训练标注掩膜；将训练图像输入至初始图像分割模型进行处理，输出预测标注掩膜；基于预测标注掩膜中像素点的预测标注、训练标注掩膜中像素点的训练标注和各种像素类型对应的基准标注，确定预测标注掩膜与训练标注掩膜之间的误差损失；基于误差损失，对每个注意力模块对应的提示图像和提示标注掩膜进行调整，得到目标图像分割模型。采用本方法，能够提高图像分割模型训练的效率。