轨迹预测模型的训练方法、装置和介质

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
轨迹预测模型的训练方法、装置和介质
申请号:CN202511149887
申请日期:2025-08-18
公开号:CN120656003B
公开日期:2025-11-25
类型:发明专利
摘要
本发明公开了一种轨迹预测模型的训练方法、装置和介质。包括:根据初始轨迹预测模型对训练样本进行预测,得到多个类别的轨迹预测结果;根据多个类别的轨迹预测结果确定目标导向损失、碰撞规避损失和轨迹分类损失;根据目标导向损失、碰撞规避损失和轨迹分类损失确定目标损失;根据样本集中训练样本对应的目标损失对初始轨迹预测模型进行迭代参数调整,得到训练完成的目标轨迹预测模型。本发明通过目标导向损失将将终点的目标方向作为轨迹规划的约束条件;通过碰撞规避损失将除障碍物以外的可通行区域作为轨迹规划的约束条件,以及通过轨迹分类损失将分类真值轨迹作为轨迹规划的约束条件,提高根据轨迹预测模型进行轨迹预测的准确性。
技术关键词
轨迹预测模型 障碍物位置信息 语义分割模型 预测类别 参数 终点 可读存储介质 计算机 标签 样本 训练装置 规划 模块 策略 数值 指令 指数