Skip to content

karolmajek/TrafficSignsRecognition

Repository files navigation

Udacity Self-Driving Car Engineer Nanodegree

Deep Learning

Project: Build a Traffic Sign Recognition Program

Overview

My neural network model trained to decode traffic signs from natural images by using the German Traffic Sign Dataset.

Click here to see full source code with visualizations in Jupyter notebook!

Dataset

Dataset - 43 categories

  1. Download the dataset. This is a pickled dataset in which we've already resized the images to 32x32.
  2. Original notebook from Udacity:
git clone https://github.com/udacity/CarND-Traffic-Signs
cd CarND-Traffic-Signs
jupyter notebook Traffic_Signs_Recognition.ipynb

Data augmentation

I have used Vivek Yadav's transform image function. It applies random affine transformation using angle,shear and translation ranges. Here is sample what generated images look like

Generated data

Images count per label after adding generated images

Generating data - images per class

Training

Training

Results on new data

Predictions

Dependencies

This project requires Python 3.5 and the following Python libraries installed:

Details and implementation

Click here to see full source code with visualizations in Jupyter notebook!