/traj_pred_vae

Primary LanguageJupyter Notebook

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

    python3 preprocess.py

Open and run all cells from traj_pred_vae.ipynb notebook