技术特征：
1.一种基于目标局部终点与多头注意力机制的轨迹预测方法，其特征在于，具体包括以下步骤：步骤1：获取数据并整理，得到数据集；建立训练集和测试集；步骤2：根据步骤1得到的训练集中的目标的历史轨迹及实际局部终点位置信息，预测目标局部终点位置；步骤3：根据步骤2得到的预测目标局部终点位置信息，采用社会池化操作提取社交信息，得到特征向量x
k
；步骤4：将步骤3得到的特征向量x
k
输入长短期记忆网络得到目标的预测轨迹；步骤5，将步骤1得到的训练集通过步骤2、3、4进行训练得到轨迹预测模型，并使用测试集对轨迹预测模型进行测试，如果不满足要求则继续迭代训练，直至得到训练好的轨迹预测模型；步骤6，将待检测的一系列图像或待检测的视频输入训练好的轨迹预测模型，得到目标的预测轨迹。2.如权利要求1所述的目标局部终点与多头注意力机制的轨迹预测方法，其特征在于，所述步骤1中，所述数据集采用stanford drone斯坦福无人机数据集。3.如权利要求1所述的目标局部终点与多头注意力机制的轨迹预测方法，其特征在于，所述步骤1中，所述步骤2包括如下子步骤：步骤21：将步骤1得到的训练集中的第k个目标实际终点位置和该目标的历史轨迹分别通过终点编码器和轨迹编码器生成对应的特征信息；其中，i表示时间点，t
p
表示预测终点之前的一个时间点，t
p
＝8，t
f
表示预测的时间终点；步骤22：将步骤21得到的特征信息组合拼接送入条件变分自编码器的潜在编码器进行训练，得到潜在变量的关键变量μ,σ；步骤23：从高斯分布中随机取样得到可能的局部终点特征z，再与步骤21中轨迹编码器生成的特征信息进行融合之后送入条件变分自编码器的潜在解码器，得到预测的目标局部终点位置4.如权利要求1所述的目标局部终点与多头注意力机制的轨迹预测方法，其特征在于，所述步骤1中，所述步骤3包括如下子步骤：步骤31：将步骤2得到的预测目标局部终点位置信息送入终点编码器，将得到的特征与步骤21中轨迹编码器生成的特征信息进行融合之后得到特征向量；步骤32：将步骤31得到的特征向量送入三个全连接层网络通过多次训练得到三个不同的全连接层网络权重，从中提取出特征向量q、k、v；步骤33：将特征向量q、k、v进行融合，融合之后的特征向量x
k
中包含了每一个目标从其他邻居目标提取的社交信息。5.如权利要求1所述的目标局部终点与多头注意力机制的轨迹预测方法，其特征在于，所述步骤33中，将特征向量q、k、v采用下式进行融合：其中，dk为特征向量q、k、v的维度，目的是为了防止训练过程中梯度消失。mij为掩码矩
阵，负责标记和预测目标在时间上和空间上有关系的目标。6.如权利要求1所述的目标局部终点与多头注意力机制的轨迹预测方法，其特征在于，所述步骤1中，所述掩码矩阵m
ij
的公式如下：其中，t
dist
为人为设置的距离阈值，表示目标对应的帧号。7.如权利要求6所述的目标局部终点与多头注意力机制的轨迹预测方法，其特征在于，所述距离阈值为100。8.如权利要求1所述的目标局部终点与多头注意力机制的轨迹预测方法，其特征在于，所述步骤1中，所述步骤4具体包括如下子步骤：步骤41：将步骤3得到的特征向量x
k
中时间维度上时刻t对应的向量x
t
作为当前时刻的输入值，当t＝1时，将当前时刻的输入值作为前一时刻隐状态h
t-1
；步骤42：将当前时刻的输入值x
t
与前一时刻隐状态h
t-1
通过公式(4)～(6)得到f
t
、i
t
、f
t
＝σ(w
f
·
[h
t-1
，x
t
]+b
f
)
ꢀꢀ
(4)i
t
＝σ(w
i
·
[h
t-1
，x
t
]+b
i
)
ꢀꢀ
(5)其中，w和b表示对应不同门网络的权重和偏置，w
f
为遗忘门权重矩阵，w
i
为输入门权重矩阵，w
c
为细胞状态权重矩阵，b
f
为遗忘门偏置，b
i
为输入门偏置，b
c
为细胞状态偏置，h
t-1
表示前一时刻隐状态的输出，σ表示网络中的激活函数是sigmoid函数，tanh表示网络中的激活函数是tanh函数；步骤43：通过公式(7)和(8)得到当前时刻t的隐状态h
t
。o
t
＝σ(w
o
[h
t-1
，x
t
]+b
o
)
ꢀꢀ
(7)h
t
＝o
t
*tanh(c
t
) (8)其中，o
t
为网络的输出门，c
t
为记忆信息；步骤44：判断当前时刻t是否为最后时刻t
f
，是则将步骤43得到的当前时刻t的隐状态h
t
作为目标在t
p
时刻到t
f
时刻的预测轨迹否则，令t＝t+1，返回步骤42。
技术总结
本发明公开了一种基于目标局部终点与多头注意力机制的轨迹预测方法：步骤1：获取数据集；步骤2：根据目标的历史轨迹及实际局部终点位置信息，预测目标局部终点位置；步骤3：采用社会池化操作提取社交信息，得到特征向量；步骤4：将特征向量输入长短期记忆网络得到目标的预测轨迹；步骤5，进行迭代训练得到训练好的轨迹预测模型；步骤6，将待检测的图像或视频输入训练好的轨迹预测模型，得到目标的预测轨迹。本发明实现了在复杂交通场景下通过特定信息丰富目标轨迹特征进行精确高效的行人轨迹预测。预测。预测。
技术研发人员：刘占文 李文倩 王超 李超 王洋 齐明远 丁璠 谭华春 孙士杰 赵彬岩 员惠莹 杨楠 贾晓航
受保护的技术使用者：长安大学
技术研发日：2022.07.21
技术公布日：2022/11/22