Skip to content
/ RRB Public

Official implementation of "Injecting Knowledge in Data-driven Vehicle Trajectory Predictors", Published in Transportation research part C.

License

Notifications You must be signed in to change notification settings

vita-epfl/RRB

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

34 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

RRB

The official code for the paper: "Injecting Knowledge in Data-driven Vehicle Trajectory Predictors", Published in Transportation research part C, 2021. Webpage , Paper , arXiv

 

drawing

Abstract:

Vehicle trajectory prediction tasks have been commonly tackled from two distinct perspectives: either with knowledge-driven methods or more recently with datadriven ones. On the one hand, we can explicitly implement domain-knowledge or physical priors such as anticipating that vehicles will follow the middle of the roads. While this perspective leads to feasible outputs, it has limited performance due to the difficulty to hand-craft complex interactions in urban environments. On the other hand, recent works use data-driven approaches which can learn complex interactions from the data leading to superior performance. However, generalization, i.e., having accurate predictions on unseen data, is an issue leading to unrealistic outputs. In this paper, we propose to learn a "Realistic Residual Block" (RRB), which effectively connects these two perspectives. Our RRB takes any off-the-shelf knowledge-driven model and finds the required residuals to add to the knowledge-aware trajectory. Our proposed method outputs realistic predictions by confining the residual range and taking into account its uncertainty. We also constrain our output with Model Predictive Control (MPC) to satisfy kinematic constraints. Using a publicly available dataset, we show that our method outperforms previous works in terms of accuracy and generalization to new scenes.

Installation

virtualenv -p /usr/bin/python3.6 rrb_env
source rrb_env/bin/activate
pip install -e trajnetbaselines/
pip install -e trajnettools/
pip install -e trajnetdataset/

Model Training

You can specify code parameters in the bash.sh file. To train the network, simply run:

bash run.sh

Model Evaluation

cd trajnetbaselines
python -m trajnetbaselines.eval --model-add <add-to-model>

You can evaluate the pre-trained models available in this repo with commands like this:

python -m trajnetbaselines.eval --model-add 'output/final_models/RRB/RRB_M_sceneGeneralization'

Citation:

@article{bahari2021injecting,
title = {Injecting knowledge in data-driven vehicle trajectory predictors},
journal = {Transportation Research Part C: Emerging Technologies},
volume = {128},
pages = {103010},
year = {2021},
issn = {0968-090X},
doi = {https://doi.org/10.1016/j.trc.2021.103010},
url = {https://www.sciencedirect.com/science/article/pii/S0968090X21000425},
author = {Mohammadhossein Bahari and Ismail Nejjar and Alexandre Alahi},
}

contact:

mohammadhossein.bahari@epfl.ch

About

Official implementation of "Injecting Knowledge in Data-driven Vehicle Trajectory Predictors", Published in Transportation research part C.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published