【技术实现步骤摘要】
轨迹预测模型训练方法、装置、设备和轨迹预测方法
[0001]本申请涉及及其学习
，特别是涉及一种轨迹预测模型训练方法、装置、设备和轨迹预测方法。
技术介绍
[0002]随着人工智能技术的迅速发展，行人轨迹预测在多个领域中受到越来越多的关注，例如自动驾驶汽车、智能交通、智慧城市等领域中均广泛的应用了行人轨迹预测技术。
[0003]在现有的行人轨迹预测方法中，通常根据获取到的各个节点之间的关系进行轨迹预测。但在现有的行人轨迹预测方法中，无法利用各个节点的特征信息，例如经纬度信息，时间信息，当前时刻人体的动作等均未被使用。借助行人轨迹预测方法无法处理得到各个特征信息之间的相关性。因此，实际应用过程中，尤其是在长期预测的场景下往往无法达到预期的预测效果。
[0004]针对相关技术中存在如何进一步提高轨迹预测的准确率，目前还没有提出有效的解决方案。
技术实现思路
[0005]基于此，有必要针对上述技术问题，提供一种能够提高行人轨迹预测的准确率的轨迹预测模型训练方法、装置、设备和轨迹预测方法。
【技术保护点】
【技术特征摘要】
1.一种轨迹预测模型训练方法，其特征在于，所述方法包括：获取初始数据，所述初始数据至少包括目标对象信息以及环境信息；基于时间属性将所述初始数据转化为有向图数据，所述有向图数据包括节点向量和有向边向量；基于每个所述节点向量的相邻节点向量确定每个所述节点向量的权重，得到训练数据集，所述训练数据集包括多个时刻的轨迹数据；基于所述训练数据集对初始模型进行训练，得到目标轨迹预测模型。2.根据权利要求1所述的方法，其特征在于，所述环境信息包括地理位置信息，时间信息和区域环境信息，所述基于时间属性将所述初始数据转化为有向图数据集包括：基于时间属性将所述初始数据转化为数据链；基于所述数据链的节点特征信息和有向边特征信息将所述数据链转化为有向图数据集，其中所述节点特征信息确定节点向量，所述有向边特征信息确定有向边向量。3.根据权利要求1所述的方法，其特征在于，所述基于每个所述节点向量的相邻节点向量确定每个所述节点向量的权重包括：采用注意力机制，基于每个所述节点向量的相邻节点向量确定每个所述节点向量的权重。4.根据权利要求1所述的方法，其特征在于，所述基于所述训练数据集对初始模型进行训练，得到目标轨迹预测模型包括：将t时刻的轨迹数据作为输入，将t+1时刻的轨迹数据作为参考数据，对所述目标轨迹预测模型进行训练。5.根据权利要求1所述的方法，其特征在于，所述基于所述训练数据集对初始模型进行训练之前包括：以任一所述节点向量为起点，沿预设方向行走预设步数，得到参...
【专利技术属性】
技术研发人员：万振弘，税雪飞，
申请(专利权)人：杭州安恒信息技术股份有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。