一种轨迹预测模型训练样本的修正方法、装置及存储介质制造方法及图纸

本发明专利技术公开一种轨迹预测模型训练样本的修正方法、装置及存储介质，包括获取训练样本、获取追踪轨迹数据、解析得到所有轨迹点、计算轨迹点的曲率及相邻轨迹点的曲率变化量，筛选出存在异常轨迹点的训练样本进行插值修复。本发明专利技术轨迹预测模型训练样本的修正方法，可以得到更平滑的追踪轨迹，提高训练样本的质量，使得模型在训练时不会拟合错误的真值，减少由于训练样本自身质量问题引发的模型预测问题，从而提升轨迹预测的准确性和稳定性。而提升轨迹预测的准确性和稳定性。而提升轨迹预测的准确性和稳定性。

【技术实现步骤摘要】
一种轨迹预测模型训练样本的修正方法、装置及存储介质


[0001]本专利技术属于无人车轨迹预测领域，具体涉及轨迹预测模型的训练样本修复方法。

技术介绍

[0002]基于深度学习模型的轨迹预测技术是目前主流的轨迹预测技术。轨迹预测模型的训练样本主要有两大类表示方法，一种是数据以图像或者点云作为训练样本，另一种是先对道路交通参与者的轨迹及地图元素进行矢量化，如waymo行为预测算法VectorNet，通过图数据结构存储作为训练样本。相较于前者，矢量化的数据解释性更强，有利于构造特征，同时对于模型来说计算成本更小。因此，本文主要研究的是使用矢量化训练样本的深度学习模型。
[0003]对于轨迹预测模型而言，训练数据主要由三部分构成：待预测车辆(Agent)，待预测车辆周围障碍物(Actor)，地图元素(lane graph)。对于预测任务来说，一般选取待预测车辆与待预测车辆周围障碍物历史x秒轨迹及待预测车辆周围的地图元素作为输入，选取待预测车辆未来y秒轨迹作为真值（GT，Ground Truth），来训练神经网络。在轨迹预测模型训练中发现，预测轨迹真值会出现曲率过大或突变等异常情况。这种轨迹会导致下游模块的误判，造成点刹或者接管，影响无人车表现。

技术实现思路

[0004]为了解决训练样本真值出现曲率过大或突变导致轨迹预测模型在训练时拟合错误真值的问题，本专利技术提出一种方法，通过对训练样本进行修正，提升轨迹预测模型的稳定性。
[0005]为实现上述目的，本专利技术实施例提供一种轨迹预测模型训练样本的修正方法，其特征在于包括：从轨迹预测训练样本集中获取训练样本；从训练样本中获取追踪轨迹数据；从追踪轨迹数据中解析得到轨迹真值对应的所有轨迹点；遍历所有轨迹点，计算轨迹点的曲率及相邻轨迹点的曲率变化量；筛选出轨迹点曲率绝对值大于曲率阈值或轨迹点曲率存在突变的训练样本作为待修复训练样本；删除待修复训练样本对应追踪轨迹中的异常轨迹点；对删除的异常轨迹点进行插值修复。
[0006]为实现上述目的，本专利技术实施例还提供一种轨迹预测模型训练样本的修正装置，其特征在于包括：训练样本提取模块，用于从轨迹预测训练样本集中获取训练样本；追踪轨迹数据提取模块，用于从训练样本中获取追踪轨迹数据；轨迹解析模块，用于从追踪轨迹数据中解析得到轨迹真值对应的所有轨迹点；
第一计算模块，用于遍历所有轨迹点，计算轨迹点的曲率；第二计算模块，用于遍历所有轨迹点，计算相邻轨迹点的曲率变化量；筛选模块，用于筛选出轨迹点曲率绝对值大于曲率阈值或轨迹点曲率存在突变的训练样本；删除模块，用于删除待修复训练样本对应追踪轨迹中的异常轨迹点；修复模块，对删除的异常轨迹点进行插值修复。
[0007]为实现上述目的，本专利技术实施例还提供一种装置，包括存储器、处理器、存储在所述存储器上并可在所述处理器上运行的程序，其特征在于所述程序被所述处理器执行时实现所述的轨迹预测模型训练样本的修正方法的步骤。
[0008]为实现上述目的，本专利技术实施例还提供一种存储介质，其特征在于所述存储介质存储有至少一个程序，所述至少一个程序可被至少一个处理器执行，以实现所述的轨迹预测模型训练样本的修正方法的步骤。
[0009]本专利技术有益效果：本专利技术的轨迹预测模型训练样本的修正方法，针对预测轨迹真值曲率发生异常的训练样本，采用插值的方法对异常轨迹点进行位置、速度、形状、加速度等特征的修正，修复错误的真值，提升模型的线上表现，减少由于训练样本自身质量引发的模型预测问题，提升轨迹预测的准确性和稳定性。
附图说明
[0010]图1为实施例1的轨迹预测模型训练样本的修正方法流程图；图2为实施例2的轨迹预测模型训练样本的修正装置流程图。
具体实施方式
[0011]下面将参照附图更详细地描述本专利技术的实施例。应当理解的是，本申请的附图及实施例仅用于示例性作用，并非用于限制本申请的保护范围。
[0012]实施例1本实施例提供一种轨迹预测模型训练样本的修正方法，如图1所示，包括：步骤101，从基于深度学习的轨迹预测模型的训练样本集中获取训练样本。训练样本主要由以下数据构成：待预测车辆与待预测车辆周围障碍物历史x秒轨迹、待预测车辆周围的地图元素及待预测车辆未来y秒轨迹，其中待预测车辆与待预测车辆周围障碍物历史x秒轨迹、待预测车辆周围的地图元素作为轨迹预测模型的输入，待预测车辆未来y秒轨迹作为轨迹预测模型的真值输出。
[0013]通过接口 get_agent_ids()，可以获得数据库中所有训练样本的追踪轨迹ID，追踪轨迹ID与训练样本一一对应，从而获取对应的训练样本。
[0014]步骤102，从训练样本中获取追踪轨迹数据，即待预测车辆未来y秒轨迹数据。
[0015]通过向接口 get_track_by_id() 输入追踪轨迹ID，可以从数据库中返回训练样本对应的追踪轨迹数据。
[0016]步骤103，从训练样本的追踪轨迹数据中，利用现有解析方法解析得到轨迹真值对应的所有轨迹点。
[0017]步骤104，遍历所有轨迹点，计算每个轨迹点的曲率，并计算每两个相邻轨迹点的曲率变化量。其中，各轨迹点的曲率计算以每相邻3个轨迹点为一组确定一个圆进行计算，两个相邻轨迹点的曲率相减取绝对值得到该两个相邻轨迹点的曲率变化量。
[0018]步骤105，将计算得到的每个轨迹点的曲率分别与预设的曲率阈值进行比较，并将计算的每相邻两个轨迹点的曲率变化量与预设的曲率变化量阈值进行比较，标记异常轨迹点，即曲率绝对值大于曲率阈值的轨迹点或曲率存在突变的轨迹点（即曲率变化量大于曲率变化量阈值的相邻两个轨迹点中的后一个轨迹点），并筛选出异常轨迹点对应的训练样本作为待修复训练样本。
[0019]步骤106，将筛选出的待修复训练样本对应追踪轨迹中的异常轨迹点删除。
[0020]步骤107，对删除的异常轨迹点进行插值修复。具体插值修复方法包括：当删除的异常轨迹点为单轨迹点，且其前、后轨迹点均为正确轨迹点时，对该异常轨迹点的前、后轨迹点的真值取平均进行插值修复，从而对异常轨迹点的位置、速度、形状、加速度等特征进行修正。
[0021]当删除的异常轨迹点为连续多个轨迹点，且其前、后轨迹点均为正确轨迹点时，计算待修复轨迹点的个数，确定步长，对该连续的多个异常轨迹点的前、后轨迹点的真值取平均，按照步长对各个轨迹点进行插值修复，从而对异常轨迹点的位置、速度、形状、加速度等特征进行修正。
[0022]当删除的异常轨迹点为单轨迹点，且其前、后轨迹点正确真值缺失，通过外插值的方法进行插值修复，从而对异常轨迹点的位置、速度、形状、加速度等特征进行修正。
[0023]当删除的异常轨迹点为连续多个轨迹点，且其前、后轨迹点正确真值缺失，将相应训练样本删除。
[0024]步骤108，获取下一个训练样本，重复上述步骤102
‑
107，直到遍历完所有训练样本。
[0025]曲率阈值和曲率变化量阈值可以根据轨迹追踪的期望目标进行本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种轨迹预测模型训练样本的修正方法，其特征在于包括：从轨迹预测训练样本集中获取训练样本；从训练样本中获取追踪轨迹数据；从追踪轨迹数据中解析得到轨迹真值对应的所有轨迹点；遍历所有轨迹点，计算轨迹点的曲率及相邻轨迹点的曲率变化量；筛选出轨迹点曲率绝对值大于曲率阈值或轨迹点曲率存在突变的训练样本作为待修复训练样本；删除待修复训练样本对应追踪轨迹中的异常轨迹点；对删除的异常轨迹点进行插值修复。2.如权利要求1所述的轨迹预测模型训练样本的修正方法，其特征在于：当删除的异常轨迹点为单轨迹点，且其前、后轨迹点均为正确轨迹点时，对所述前、后轨迹点的真值取平均进行插值修复。3.如权利要求1所述的轨迹预测模型训练样本的修正方法，其特征在于：当删除的异常轨迹点为连续多个轨迹点，且其前、后轨迹点均为正确轨迹点时，计算待修复轨迹点的个数，确定步长，对所述前、后轨迹点的真值取平均，按照步长对各个轨迹点进行插值修复。4.如权利要求1所述的轨迹预测模型训练样本的修正方法，其特征在于：当删除的异常轨迹点为单轨迹点，且其前、后轨迹点正确真值缺失，通过外插值的方法进行插值修复。5.如权利要求1所述的轨迹预测模型训练样本的修正方法，其特征在于：当删除的异常轨迹点为连续多个轨迹点，且其前、后轨迹点正确真值缺失，将相应训练样本删除。6.一种轨迹预测模型训练样本的修正装置，其特征在于包括：训练样本提取模块，用于从轨迹预测训练样本集中获取训练样本；追踪轨迹数据提取模块，用于从训练样本中获取追踪轨迹数据；轨迹解析模块，用于从追踪轨迹数据中解析得到轨迹...

【专利技术属性】
技术研发人员：朱众志，徐培治，郭谢，
申请(专利权)人：中智行科技有限公司，
类型：发明
国别省市：