基于先验信息的指代图像分割模型训练方法、装置及设备

本发明涉及人工智能技术领域，提供一种基于先验信息的指代图像分割模型训练方法、装置及设备。本发明通过获取样本训练集，根据训练图片和文本指代描述，获取训练图片的先验引导热图，根据多个先验引导热图和样本训练集对初始指代图像分割模型进行训练，并根据预测掩码标注数据、掩码标注数据以及损失函数调整模型的权重参数，直至模型收敛，获取训练完成的指代图像分割模型。由于先验引导热图包含掩码标注数据的先验信息，在通过先验引导热图和样本训练集对模型进行训练过程中，能够使得模型快速关注到指代目标对应的区域，且更好的融合图像细节信息与上下文语义指代信息，提高对训练图片中指代目标的定位的准确性，从而提高了对指代目标的分割效果。