Udacity Self Driving Car Capstone

This repo contains the scripts for training the traffic light detector for the capstone project.

Run install_ubuntu.sh

You may want to run this script in a virtualenv.

requirements.txt contains tensorflow 1.4.1

./train.sh <ssd_inception | ssd_mobilenet>

./freeze.sh <ssd_inception | ssd_mobilenet> <checkpoint>

If you have trained the model a version of tensorflow > 1.4.1 you can create a virtualenv with requirements.txt installed and run this script.

inference.ipynb

swarmt/CarND-Traffic-Light-Detection