Skip to content

sebastianfernandezgarcia/Traffic-Signs-Recognition-Machine-Learning

Repository files navigation

Traffic Signs Recognition ⚠️🚦

Introduction 📄

Development of a neural network capable of recognizing nine different danger warning traffic signs.

  1. Sign P-50. Other dangers.
  2. Signal P-14b. Dangerous curves to the left.
  3. Sign P-20. Pedestrians.
  4. Sign P-19. Slippery pavement.
  5. Sign P-18. Works.
  6. Signal P-3. Traffic lights.
  7. Sign P-15a. Highlight.
  8. Sign P-1. Intersection with priority.
  9. Signal P-24. Passage of animals in freedom.

Screenshot 2022-01-09 at 20 48 42

Datasets 📁

The images, both for the training set and for the validation set, have been extracted from several datasets, the most notable being the GTSRB (German Traffic Sign Recognition Benchmark) dataset: https://www.kaggle.com/meowmeowmeowmeowmeow/gtsrb -german-traffic-sign

In addition, self-made images have been added.

Data Augmentation

Data augmentation has been performed on the training set, to obtain new images from those already present. To do this, we have used various parameters such as rotation angle, cut angle, among others, in the ImageDataGenerator class

Additional Information

All the implementation details are in the memory that we have made, which can be read from the following link: Proyect Memory

In this document, the hyperparameters that have been used are detailed, as well as those that we have tested. Also, the concept of Categorical Cross Entropy is explained, and how this loss function works.

Authors ✒️

Base code provided by Cayetano Guerra Artal

About

IA for recognising Traffic Signs using Keras [Tensorflow]

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published