轨迹预测模型的训练方法以及轨迹预测方法

本公开提供了一种轨迹预测模型的训练方法以及轨迹预测方法，涉及自动驾驶和智能交通等人工智能技术领域。其中，初始轨迹预测模型包括：嵌入层、编码器和解码器，该方法包括：利用嵌入层对样本车辆的历史轨迹信息以及与历史轨迹信息关联的局部地图信息进行特征提取，得到车辆轨迹特征和地图特征；利用编码器对车辆轨迹特征和地图特征进行编码，得到车辆轨迹编码和地图元素编码；利用解码器对车辆轨迹编码和地图元素编码进行解码，得到目标车辆的目标预测轨迹；根据目标预测轨迹和目标车辆的真实行驶轨迹之间的损失对初始轨迹预测模型进行训练，得到轨迹预测模型。本公开提供的方法提升了轨迹预测模型的预测准确性以及轨迹预测模型的存储性能。