midemig / traj_pred_vae

Vehicles trajectory prediction using Recurrent VAE network

Tensorflow implementation of the method presented in "Vehicles trajectory prediction using Recurrent VAE network"

Dataset

The dataset used for these experiments can be found on https://www.highd-dataset.com/

Requirements

• Python 3.7.8
• TensorFlow 2.4.1
• Keras 2.4.0
• Pandas 1.2.1
• Numpy 1.19.5

Usage

Extract all the dataset csv into data/ folder

    python3 extract.py
    python3 preprocess.py

Open and run all cells from traj_pred_vae.ipynb notebook