Top AI Repos — open-source AI, indexed and scored
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Top AI Repos tracks AI repositories on GitHub and answers two different questions about each one: is it moving right now, and would you bet a product on it.
Code for "Social-STGCNN: A Social Spatio-Temporal Graph Convolutional Neural Network for Human Trajectory Prediction" CVPR 2020
| Date | Stars |
|---|---|
| 2026-07-31 | 586 |
| 2026-08-01 | 586 |
| 2026-08-02 | 586 |
| 2026-08-05 | 587 |
| 2026-08-06 | 587 |
Today
— stars today
This week
— stars this week
This month
— stars this month
Momentum
0.0
growth rate 0.00%/day
# Social-STGCNN: A Social Spatio-Temporal Graph Convolutional Neural Network for Human Trajectory Prediction
### Abduallah Mohamed, Kun Qian <br />
### Mohamed Elhoseiny** , Christian Claudel**
##### <sup>**</sup> Equal advising
Read full paper <a href="https://arxiv.org/abs/2002.11927">here</a>, Presented at [CVPR 2020](http://cvpr2020.thecvf.com/)
### Check our latest work in motion prediction with new metrics (ECCV2022): Social-Implicit https://github.com/abduallahmohamed/Social-Implicit
### Social-STGCNN
<div align='center'>
<img src="images/Social-STGCNN-Teaser_.jpg"></img>
</div>
<br />
We propose the Social Spatio-Temporal Graph Convolutional Neural Network (Social-STGCNN), which models the problem of human trajectory prediction as a spatio-temporal graph. Our results show an improvement over the state of art by 20\% on the Final Displacement Error (FDE) and an improvement on the Average Displacement Error (ADE) with 8.5 times less parameters and up to 48 times faster inference speed than previously reported methods. In addition, our model is data efficient, and exceeds previous state of the art on the ADE metric with only 20\% of the training data. We propose a kernel function to embed the social interactions between pedestrians within the adjacency matrix.
Our model inference speed is 0.002s/frame (500Hz) using only 7.6K parameters.
### Citaion
You can cite our paper using:
```
@inproceedings{mohamed2020social,
title={Social-STGCNN: A Social Spatio-Temporal Graph Convolutional Neural Network for Human Trajectory Prediction},
author={Mohamed, Abduallah and Qian, Kun and Elhoseiny, Mohamed and Claudel, Christian},
booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
pages={14424--14432},
year={2020}
}
```
#### Model
<div align='center'>
<img src="images/social-stgcnn-model.jpg"></img>
</div>
<br />
Social-STGCNN model consists of 2 building blocks: <br />
1- ST-GCNN: A Spatio-Tempral Graph CNN that creates a spatio-temporal graph embedding representing the previous pedestrians trajectories. <br />
2- TXP-CNN: A Time-Extrapolator CNN that utilizes the spatio-temporal graph embedding to predict future trajectories.<br />
More details are in this description video:<br />
<div align='center'>
[](https://www.youtube.com/watch?v=Yi9rVHJM8WY)
</div>
#### Predictions sample
<div align='center'>
<img src="images/social-stgcnn-pred.gif"></img>
</div>
<br />
### Setup:
The code was written using python 3.6.
The following libraries are the minimal to run the code:
```python
import pytorch
import networkx
import numpy
import tqdm
```
or you can have everything set up by running:
```bash
pip install -r requirements.txt
```
### Using the code:
To use the pretrained models at `checkpoint/` and evaluate the models performance run:
```bash
test.py
```
To train a model for each data set with the best configuration as in the paper, simply run:
```bash
./train.sh
```
Please note: The initiation of training and testing might take a while as the code creates a cache to store the spatio-temporal graphs.
<br />
Exact CVPR version: https://github.com/abduallahmohamed/Social-STGCNN/tree/ebd57aaf34d84763825d05cf9d4eff738d8c96bb
### Check our work in the 3D motion prediction (ICCV2021): https://github.com/abduallahmohamed/Skeleton-Graph
Excerpt of 3,426 characters
Read on GitHub
Abduallah Mohamed · Meta
27
Would you bet a product on this? Bounded 0–100 and slow moving.
matched fp:7ec88c2c904daad7, llm:Repository topics and description: 'gcnn, graph-convolutional-networks, graph-neural-networks, human-trajectory-prediction, pedestrian-trajectories, realtime, social-stgcnn, spatio-temporal-graphs' and description: 'Code for "Social-STGCNN: A Social Spatio-Temporal Graph Convolutional Neural Network for Human Trajectory Prediction" CVPR 2020'.
matched fp:7ec88c2c904daad7, llm:Repository topics and description: 'gcnn, graph-convolutional-networks, graph-neural-networks, human-trajectory-prediction, pedestrian-trajectories, realtime, social-stgcnn, spatio-temporal-graphs' and description: 'Code for "Social-STGCNN: A Social Spatio-Temporal Graph Convolutional Neural Network for Human Trajectory Prediction" CVPR 2020'.
matched fp:7ec88c2c904daad7, llm:Repository topics and description: 'gcnn, graph-convolutional-networks, graph-neural-networks, human-trajectory-prediction, pedestrian-trajectories, realtime, social-stgcnn, spatio-temporal-graphs' and description: 'Code for "Social-STGCNN: A Social Spatio-Temporal Graph Convolutional Neural Network for Human Trajectory Prediction" CVPR 2020'.