/GTSRB

Convolutional Neural Network for German Traffic Sign Recognition Benchmark

Primary LanguageJupyter NotebookMIT LicenseMIT

Overview

A convolutional neural network for German traffic sign image classification.

Dataset

German Traffic Sign Recognition Dataset (GTSRB) is an image classification dataset.
The images are photos of traffic signs. The images are classified into 43 classes. The training set contains 39209 labeled images and the test set contains 12630 images. Labels for the test set are not published.
See more details here.

Model

ResNet-34 pretrained on ImageNet dataset, then finetuned on GTSRB dataset.

Deep Learning Libraries

fastai with PyTorch backend.

Metrics

The model achieved 99.22% accuracy on the validation set (random 20% subset of the training dataset).