一种轨迹预测模型的训练方法、装置、设备及介质

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种轨迹预测模型的训练方法、装置、设备及介质
申请号:CN202411503142
申请日期:2024-10-25
公开号:CN119377699A
公开日期:2025-01-28
类型:发明专利
摘要
本发明实施例公开了一种轨迹预测模型的训练方法、装置、设备及介质。其中,该方法包括:获取目标对象在当前时刻的当前环境信息和历史时刻的历史环境信息;将当前环境信息和历史环境信息分别输入至轨迹预测模型中,得到当前预测轨迹和历史预测轨迹;根据当前预测轨迹和历史预测轨迹确定相似度损失,相似度损失用于描述当前预测轨迹与历史预测轨迹之间的差异;基于相似度损失对轨迹预测模型进行训练,以基于训练完成的轨迹预测模型对目标对象进行运动轨迹预测。本方案基于根据当前预测轨迹和历史预测轨迹确定的相似度损失对轨迹预测模型进行训练,能够有效地提高轨迹预测在时序上的连贯性和稳定性,有助于提高自动驾驶系统的安全性和可靠性。
技术关键词
轨迹预测模型 地图特征 特征提取单元 轨迹特征 运动轨迹预测 对象 融合特征 车道 自动驾驶系统 特征金字塔 模型训练模块 可读存储介质 计算机 线性单元 训练装置 电子设备 处理器通信