Skip to content
/ GTSRB Public

Convolutional Neural Network for German Traffic Sign Recognition Benchmark

License

Notifications You must be signed in to change notification settings

surmenok/GTSRB

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 

Repository files navigation

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).