/traffic-signs

Classify Traffic Signs.

Primary LanguageJupyter Notebook

Self-Driving Car Engineer Nanodegree

Deep Learning

---- 作者的话开始 ----

效果

本程序实现一个可以识别交通标志的神经网络,输入为一张32*32的彩色图像,输出为43种交通标志的预测结果。

数据集地址下载地址:traffic-sign-data.zip 百度云

数据集预览

思路

照着 Keras 的 cifar10_cnn 搭就行,训练结果 97% 左右。

我的代码:Traffic_Signs_Recognition.ipynb

参考文献:Traffic Sign Recognition with Multi-Scale Convolutional Networks

---- 作者的话结束 ----

Project: Build a Traffic Sign Recognition Program

This is a Work In Progress

Install

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

In addition to the above, for those optionally seeking to use image processing software, you may need one of the following:

For those optionally seeking to deploy an Android application:

  • Android SDK & NDK (see this README)

If you do not have Python installed yet, it is highly recommended that you install the Anaconda distribution of Python, which already has the above packages and more included. Make sure that you select the Python 3.5 installer and not the Python 2.x installer. pygame and OpenCV can then be installed using one of the following commands:

Run this command at the terminal prompt to install OpenCV:

opencv
conda install -c https://conda.anaconda.org/menpo opencv3

Run this command at the terminal prompt to install PyGame:

PyGame:
Mac: conda install -c https://conda.anaconda.org/quasiben pygame Windows: conda install -c https://conda.anaconda.org/tlatorre pygame Linux: conda install -c https://conda.anaconda.org/prkrekel pygame

Code

A template notebook is provided as Traffic_Signs_Recognition.ipynb. While no code is included in the notebook, you will be required to use the notebook to implement the basic functionality of your project and answer questions about your implementation and results.

Run

In a terminal or command window, navigate to the project directory that contains this README and run the following command:

jupyter notebook Traffic_Signs_Recognition.ipynb

This will open the Jupyter Notebook software and notebook file in your browser.

Data

  1. Download the dataset (2 options)
    • You can download the pickled dataset in which we've already resized the images to 32x32 here.
    • (Optional). You could also download the dataset in its original format by following the instructions here. We've included the notebook we used to preprocess the data here.