Skip to content

Latest commit

 

History

History

Traffic-Sign-Classifier

LeNet Implementation for Traffic Sign Classification - Tensorflow

Udacity - Self-Driving Car NanoDegree

The object of the project is to distinguish 43 different types of traffic sign used is Germany. Number of Output layers in LeNet has changed to 43. Then, the network has retrained using Traffic sign data

Dataset

Images of German Traffic Sign Dataset have used for training and test. Pickled datasets are available below.

Vanilla Pickled dataset

Augmented Pickled Dataset

or you can download all data Traffic_Signs_data.zip (395MB)

Training

Network

LeNet with Batch Normalization before each activation layer. No dropout!

Convolutional weights were initialized by 'He' method.

Environment

  • Python 3.5.2
  • Tensorflow 1.0.1

Optimizer Settings

  • Optimizer: Adam
  • Learning rate: 10e-3
  • Loss: Cross entropy
  • Batch Size: 1024
  • Epoch: 100

Result

Test Accuracy = 94.0%

Inference using arbitrary traffic sign data

Confusion Matrix