用于轨迹预测的 Transformer 网络 这是论文的代码 要求 pytorch 1.0+ 麻木 西比 熊猫 张量板 (项目中包含的是修改版) 用法 数据设置 数据集文件夹必须具有以下结构: - dataset - dataset_name - train_folder - test_folder - validation_folder (optional) - clusters.mat (For quantizedTF) 个人变压器 要训​​练,只需运行具有不同参数的train_individual.py 示例:训练 eth 的数据 CUDA_VISIBLE_DEVICES=0 python train_individualTF.py --dataset_name eth --name eth --max_epoch 240 --bat
1
出租车轨迹预测 预测葡萄牙波尔图出租车的目的地。
2021-10-29 20:20:32 87KB R
1
轨迹预测 使用机器学习算法(例如lstm,seq2seq等)预测船只轨迹的python框架。
2021-09-07 23:26:14 11KB security machine-learning navigation lstm
1