This repository contains the PyTorch code for ICPR 2020 paper:
DAG-Net: Double Attentive Graph Neural Network for Trajectory Forecasting
Alessio Monti,
Alessia Bertugli,
Simone Calderara,
Rita Cucchiara
The model is composed by three main components: the generative model (VRNN) and two graph neural networks.
The first graph network operates on agents' goals, expressed as specific areas of the environment where agents will land in the future. The second second graph network operates instead on single agents' hidden states, which contain past motion behavioural information. The employing of two separate graph neural networks allows to consider and share both past and future information while generating agents' future movements.
- Python >= 3.8
- PyTorch >= 1.5
- CUDA 10.0
- Clone this repo:
git clone https://github.com/alexmonti19/dagnet.git
cd dagnet
- Create a new virtual environment using Conda or virtualenv.
conda create --name <envname>
- Activate the environment and install the requirements:
conda activate <envname>
pip install -r requirements.txt
Our proposal is general enough to be applied in different scenarios: the model achieves state-of-the-art results in both urban environments (Stanford Drone Dataset) and sports applications (STATS SportVU NBA Dataset).
For complete information on where to download and how to preprocess the datasets see the relative datasets/README.md.
The repo contains both the final model and the two ablation architectures cited in the paper.
- VRNN: the baseline generative architecture
- A-VRNN: enhanced version of the baseline with a single graph (on agents' hidden states)
- DAG-Net: the complete architecture with two graphs
For more information on how to train the models see the relative models/README.md.
We also ship the checkpoints to reproduce the results presented in the paper.
Offense BSK | Defense BSK | Stanford Drone | ||||
---|---|---|---|---|---|---|
ADE | FDE | ADE | FDE | ADE | FDE | |
VRNN | 9,41 | 15,56 | 7,16 | 10,50 | 0,58 | 1,17 |
A-VRNN | 9,48 | 15,52 | 7,05 | 10,34 | 0,56 | 1,14 |
DAG-Net | 8,98 | 14,08 | 6,87 | 9,76 | 0,53 | 1,04 |
For more information on how to test the models and produce the above results, see the relative models/README.md.
If you have any questions, please contact alessio.monti@unimore.it or alessia.bertugli@unimore.it, or open an issue on this repo.
If you find this repository useful for your research, please cite the following paper:
@proceedings{monti2020dagnet,
title={DAG-Net: Double Attentive Graph Neural Network for Trajectory Forecasting},
author={Alessio Monti and Alessia Bertugli and Simone Calderara and Rita Cucchiara},
booktitle = {25th International Conference on Pattern Recognition (ICPR)},
year={2020}
}