本程序实现一个可以识别交通标志的神经网络,输入为一张32*32的彩色图像,输出为43种交通标志的预测结果。
数据集地址下载地址:traffic-sign-data.zip 百度云
数据集预览
照着 Keras 的 cifar10_cnn 搭就行,训练结果 97% 左右。
我的代码:Traffic_Signs_Recognition.ipynb
参考文献:Traffic Sign Recognition with Multi-Scale Convolutional Networks
This is a Work In Progress
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:
- PyGame
- Helpful links for installing PyGame:
- Getting Started
- PyGame Information
- Google Group
- PyGame subreddit
- OpenCV
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
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.
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.