一种视觉位置识别模型的训练方法和相关装置

本申请公开了一种视觉位置识别模型的训练方法和相关装置，涉及人工智能技术领域。包括：获取训练数据；通过待训练的预测模型中的自注意力模块和交叉注意力模块，根据查询图像和对比图像，生成查询图像的综合特征向量和对比图像的综合特征向量；根据查询图像的综合特征向量和对比图像的综合特征向量的差异，对待训练的预测模型进行训练，得到视觉位置识别模型。由于两个综合特征向量均关注到查询图像和对比图像的重要特征，所以基于两个综合特征向量的差异训练得到的视觉位置识别模型，可以快速确定出查询图像中需要关注的重点特征，可以更高效地从图像数据库中确定出与查询图像相似度最高的目标图像。