Skip to content

Replicating LeNet-5 convolutional neural network architecture with PyTorch to predict traffic signs using German Traffic Sign Recognition Benchmark (GTSRB) dataset.

Notifications You must be signed in to change notification settings

devdezzies/gtsrb-traffic-vision

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GTSRB Traffic Vision with LeNet-5 (modified)

An image classifier using Convolutional Neural Network (CNN) to predict German traffic signs.

alt text source

Overview dataset

German Traffic Sign Recognition Benchmark (GTSRB) is a large multi-category classification benchmark. The first was used in a competition at IJCNN 2011. The images are classified into 43 classes, consisting of 26640 training labeled images and 12630 testing set.

sample

Models

There are few trained models with different configuration and CNN architecture

  • GTSRBModelV0
  • GTSRBModelV1 (work in progress)

Sample Predictions

sample model: GTSRBModelV0

About

Replicating LeNet-5 convolutional neural network architecture with PyTorch to predict traffic signs using German Traffic Sign Recognition Benchmark (GTSRB) dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published