基于自蒸馏学习的自动驾驶轨迹预测方法、装置、及设备

本申请提供的一种基于自蒸馏学习的自动驾驶轨迹预测方法、装置、及设备，应用于自动驾驶技术领域，该方法首先获取待预测数据，将待预测数据输入至轨迹预测网络中处理，得到待预测数据对应的轨迹预测结果，其中，轨迹预测网络是样本完整轨迹数据、样本地图数据、样本掩码轨迹数据和预设损失函数对初始轨迹预测网络训练得到的；预设损失函数是基于完整轨迹损失函数、掩码轨迹损失函数及最大均值差异损失函数构建。该技术方案采用完整轨迹数据和掩码轨迹数据训练得到的轨迹预测网络，适用于不同复杂程度的轨迹分析任务，将待预测数据输入至轨迹预测网络中得到轨迹预测结果，实现了对待预测轨迹数据的自动处理与预测，提升了轨迹预测的效率。