vehicle-predictor

Downloading dataset

cd data
sh get_data.sh

Model repo

All models are found on this drive folder:

Activating Environment

conda env create -f environment.yml

Note: torch packages are installed with pip for GPU compatibilities. If you are installing locally on a machine without GPU, you might have to install accordingly as per https://pytorch.org/