Spatio Temporal Prediction Python Packages

torch-spatiotemporal

tsl: a PyTorch library for processing spatiotemporal data.

trajectron

Code accompanying the ECCV 2020 paper "Trajectron++: Dynamically-Feasible Trajectory Forecasting With Heterogeneous Data" by Tim Salzmann*, Boris Ivanovic*, Punarjay Chakravarty, and Marco Pavone (* denotes equal contribution).

geotorchai

GeoTorchAI: A Framework for Training and Using Spatiotemporal Deep Learning Models at Scale

stm-gnn

STM-Graph: A Python Framework for Spatio-Temporal Mapping and Graph Neural Network Predictions

trajectronpp

Code accompanying the ECCV 2020 paper "Trajectron++: Dynamically-Feasible Trajectory Forecasting With Heterogeneous Data" by Tim Salzmann*, Boris Ivanovic*, Punarjay Chakravarty, and Marco Pavone (* denotes equal contribution).