CarND-Traffic-Light-Classifier Dataset of Udacity CarND Capstone Project Simulator Real World CSV and model training scripts are already present Train the model and create .h5 files. Load .h5 files and predict. Please keep the python version same while training and inference otherwise you may run into Bad Marshal data error