Skip to content

Code and models for AAAI 2021 paper: *Temporal Pyramid Network for Pedestrian Trajectory Prediction with Multi-Supervision

Notifications You must be signed in to change notification settings

Blessinglrq/TPNMS

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

TPNMS

Code and models for AAAI 2021 paper: Temporal Pyramid Network for Pedestrian Trajectory Prediction with Multi-Supervision

Environment

  • Python 3.8
  • pytorch 1.11.0
  • cuda 11.3
  • Ubuntu 20.04
  • RTX 3090
  • Please refer to the "requirements.txt" file for more details.

Usage

To test the model, run:

scripts/evaluate_model.py

To train the model, run:

scripts/train_TPN_P.py

Citation

If you find this work useful in your research, please consider citing:

@inproceedings{liang2021temporal,
  title={Temporal Pyramid Network for Pedestrian Trajectory Prediction with Multi-Supervision},
  author={Liang, Rongqin and Li, Yuanman and Li, Xia and Tang, Yi and Zhou, Jiantao and Zou, Wenbin},
  booktitle={Proceedings of the AAAI Conference on Artificial Intelligence},
  volume={35},
  number={3},
  pages={2029--2037},
  year={2021}
}

Contact

If you encounter any issue when running the code, please feel free to reach us either by creating a new issue in the github or by emailing

About

Code and models for AAAI 2021 paper: *Temporal Pyramid Network for Pedestrian Trajectory Prediction with Multi-Supervision

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages