Follow the following steps to set up:
git clone https://github.com/dylantzx/Trajectory-Prediction.git
pip install -r requirements.txt # install
In this project, we are motivated by self-driving vehicles operating in a dynamic and uncertain environment. To be more specific, we focused on the urban traffic scene where the road is shared by a set of distinct agents like cars, pedestrians and cyclists. Self-driving vehicles require accurate and precise future trajectories prediction of the surrounding agents, where a trajectory is referred to as a sequence of x-y points. However, due to the stochastic nature of each agent’s future behaviour, predicting future trajectories becomes a relatively complicated process.
To enable us to model a more realistic trajectory prediction, we trained and evaluated our approach using a well-established public autonomous driving dataset, the nuScenes dataset. The team has attempted various ways to improve the evaluation results of the model, such as experimenting on different model designs, trying different normalization techniques and increasing the modality of the model by incorporating more input features. The evaluation results were then compared with those from the state-of-the-art prediction models, and have shown a reasonable evaluation score.
The goal is to predict the 3 seconds future trajecotries of vehicles (Cars, Vans and Trucks) in the nuScenes dataset.
For more details, you can read our report
Simply execute the training script to train on the trg.csv dataset:
python3 train.py
For evaluation, edit the checkpoint file obtained via training in line 210 where PATH = <your checkpoint file path>, and run the script:
python3 evaluate.py
| Approach | One Step | Look Ahead |
|---|---|---|
| Multi Task LSTM without Normalization | ADE: 14.233 FDE: 14.233 | ADE: 698.29 FDE: 698.29 |
| Normalization across whole training and validation dataset between -1 and 1 | ADE: 5.332 FDE: 5.332 | ADE: 79.999 FDE: 79.999 |
| Normalization initialized on every instance between 0 and 1 | ADE: 0.9839 FDE: 0.9839 | ADE: 4.9409 FDE: 4.9409 |
| Sequence length 4 with normalization initialized on every instance between -1 and 1 | ADE: 0.7373 FDE: 0.7373 | ADE: 3.1019 FDE: 3.1019 |
| Sequence length 6 with normalization initialized on every instance between -1 and 1 | ADE: 0.6629 FDE: 0.6629 | ADE: 2.923 FDE: 2.923 |
| Sequence length 8 with normalization initialized on every instance between -1 and 1 | ADE: 0.829 FDE: 0.829 | ADE: 4.176 FDE: 4.176 |
| Increased modality by including the z translation and heading angle | ADE: 1.3113 FDE: 1.3113 | ADE: 5.7129 FDE: 5.7129 |
| Existing Solutions | ||
| MTP | ADE: 4.42 FDE: 10.36 | |
| CoverNet | ADE: 3.87 FDE: 9.26 | |