STF-Spatial-Temporal-Fusion-for-Trajectory-Prediction

code of paper "STF: Spatial Temporal Fusion for Trajectory Prediction" This repo is based on the awesome work: GRIP++

Running

To execute the code in main.py, first run data_process.py to generate a new dataset.

dataset

The raw apollo dataset can be downloaded at the apolloscape.auto or ApolloScapeAuto/dataset-api

Citation

If you find STF paper useful in your research or applications, please kindly cite:

@article{han2023stf,
  title={STF: Spatial Temporal Fusion for Trajectory Prediction},
  author={Han, Pengqian and Roop, Partha and Liu, Jiamou and Bao, Tianzhe and Wang, Yifei},
  journal={arXiv preprint arXiv:2311.18149},
  year={2023}
}