EGO-only
|
minADE |
minFDE |
MR |
| HiVT |
1.657357096672058 |
3.016077995300293 |
0.4182955324649811 |
| V2ITrajNet |
1.2887325286865234 |
2.074836015701294 |
0.30570757389068604 |
| V2X-Graph |
1.1508910655975342 |
2.007390022277832 |
0.3013002276420593 |
- parameter setting
- batch size: 16
- gpus: 8, ddp
- epoch: 100, v2x-graph: 50
- dataset: /V2X-Seq-TFD/cooperation-vehicle-infrastructure/vehicle-trajectories
- Baselines
- HiVT (slurm)
- path: …./HiVT_plugin/
- checkpoint: ./lightning_logs/version_64/checkpoints/
- visualize: ./vis_output
- script
- train: hivt_pretrain_ego.sh
- eval: hivt_eval_ego.sh
- visualize: visualize_prediction.sh
- V2ITrajNet (slurm)
- path: …./HiVT_custom_v8
- checkpoint: ./my_logs/EGO_v0.3_2/
- visualize: ./vis_output
- script
- train: only_ego.sh
- eval: eval_only_ego.sh
- visualize: visualize_prediction.sh
- V2X-Graph (not slurm)
- path: …./V2X-Graph/
- checkpoint: ./lightning_logs/
- visualize: ./vis_output
- script
- train: train_seq_infra.sh
- eval: eval_seq_infra.sh
- visualize: visualize_prediction.sh