基于Transformer的单车轨迹预测

前言

• 采用Transformer网络架构实现单车轨迹预测
• 后续补充模型和数据的细节
• 代码可在该处找到https://github.com/limeng-1234/transformer-trajectory_prediction

数据集

数据集采用HighD。为简化，仅使用数据集中一个csv文件中的车辆轨迹数据作为建模的数据。

模型

模型采用pytorch自带的Transformer模型。可参考10分钟学会Transformer。

此外，为便于理解Transformer的建模，在开源的github中提供了由pytorch底层代码编写的Transformer实现。