Skip to content

xuehaouwa/Trajectory-Prediction-Tools

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Trajectory-Prediction-Tools

topic HitCount Build Status Version

This pytp package is a tool package for the Trajectory Prediction task. It contains tools for data augmentation, normalization, preprocessing, evaluation and so on.

Maintainer - Hao Xue

Installation

pip install pytp

Usage

  • Evaluation (calculate ADE and FDE)

    from pytp.utils.evaluate import get_ade, get_fde
    
    ade = get_ade(predicted, ground_truth)
    fde = get_fde(predicted, ground_truth)
  • Augmentation

    from pytp.utils.augment import Augmentation
    
    aug = Augmentation()

Requirements

  • Python >= 3.6

Citation

If you find this package useful, please consider citing:

@inproceedings{xue2019pedestrian,
  title={Pedestrian Trajectory Prediction Using a Social Pyramid},
  author={Xue, Hao and Huynh, Du Q and Reynolds, Mark},
  booktitle={Pacific Rim International Conference on Artificial Intelligence},
  pages={439--453},
  year={2019},
  organization={Springer}
}
@inproceedings{xue2019location,
  title={Location-Velocity Attention for Pedestrian Trajectory Prediction},
  author={Xue, Hao and Huynh, Du and Reynolds, Mark},
  booktitle={2019 IEEE Winter Conference on Applications of Computer Vision (WACV)},
  pages={2038--2047},
  year={2019},
  organization={IEEE}
}

To Do

  • Documentation
  • Unit Test coverage
  • Documentation website
  • Add tools for visualization
  • Add model zoos to include some trajectory prediction methods

About

A tool set for trajectory prediction, ready for pip install

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages