
Trajectory-Transformer:轨迹预测的变压器网络代码
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:Trajectory-Transformer是一种创新的变压器网络架构,专门用于高效准确地进行轨迹预测。该模型通过编码行人历史移动数据,利用自注意力机制捕捉复杂场景中的动态关系,为自动驾驶和智慧城市应用提供强大支持。
用于轨迹预测的Transformer网络要求使用Pytorch 1.0+版本。项目包含一个经过修改的版本。
数据设置:
数据集文件夹必须具有以下结构:
- dataset
- dataset_name
- train_folder
- test_folder
- validation_folder (可选)
- clusters.mat (适用于量化TF)
要训练个人Transformer,只需运行train_individual.py脚本,并根据需要调整参数。示例命令如下:
训练eth数据集的命令为:CUDA_VISIBLE_DEVICES=0 python train_individualTF.py --dataset_name eth --name eth --max_epoch 240 --bat
全部评论 (0)
还没有任何评论哟~


