CNN classifier to predict VW Golf models. Models currently supported:

• Golf 5
• Golf 6
• Golf 7
• Golf 7.5

For more info on the dataset and models, check out the respective Wikis.

Tested and trained on Ubuntu 18.04, Driver Version: 430.64, CUDA 10.0, Tensforflow 2.0.0 and Keras 2.3.1. See install instructions here.

Install the Python packages with:

$ (venv) pip install -r requirements.txt

To train the providided dataset, use:

$ (venv) python train.py

Note: Training script is now behind the latest model notebook "golf_classifier_final_model.ipynb".

A pretrained model is to be included.