基于注意力机制的短道速滑运动轨迹预测模型 |
| |
引用本文: | 张子涵,周斌,李文豪.基于注意力机制的短道速滑运动轨迹预测模型[J].现代计算机,2022(14):28-34. |
| |
作者姓名: | 张子涵 周斌 李文豪 |
| |
作者单位: | 中南民族大学计算机科学学院 |
| |
基金项目: | 湖北省自然科学基金项目(2016CFB650); |
| |
摘 要: | 为了解决短道速滑中多名运动员在拥挤状态下容易出现轨迹判断错误的问题,本文设计了一种基于注意力机制的轨迹预测模型。把位置和速度信息输入LSTM编码器中,再通过注意力模块对速度信息进行加权求和,最后整合速度和位置的隐藏状态输入到LSTM解码器来对短道速滑轨迹进行预测。结果采用平均位移误差(ADE)和最终位移误差(FDE)进行评估。结果表明:提出的轨迹预测模型在短道速滑运动员训练数据集中和在公开数据集中与基准模型相比平均ADE和FDE精度明显优于其他网络模型,具有一定的实用价值。
|
关 键 词: | 轨迹预测 长短期记忆(LSTM) 短道速滑 |
|
|