Skip to content

Using tensor flow to build a convolutional neural network to classify 43 different types of traffic signs

Notifications You must be signed in to change notification settings

drmerlot/Traffic-Sign-Classifier

Repository files navigation

Classifying Traffic Signs with a Convolutional Neural Network and Tensorflow

Overview

This repository demonstrates using tensorflow to build and train a simple convolutional neural network which can classify 43 different types of traffic signs. The project was completed as part of the Udacity Self Drving Car nanodegree program. The resulting network was trained to an accuracy of 98% and performed with an accuracy of 92% on holdout test data. In addition, the network was able to correctly classify picutres of signs not included in training or test data that were partially obscured by grafiti or snow. In addition, some extra pictures were fed though the network to see how it would respond to images far outside the training bounds.

Some examples:

  • The network correctly classifed this image as "speed limit 30 kph" despite being spray painted:

  • And this image correctly as "yeild" even though it is partially covered by snow:

  • And apparently there's a 59% chance I'm a speed limit sign See the resemblance?:

Launch the Ipython notebook here:

https://github.com/andrewsommerlot/Traffic-Sign-Classifier/blob/master/traffic-sign-classifier.ipynb