This project focuses on building a deep learning model to classify traffic signs using Convolutional Neural Networks (CNNs). The model is trained on the German Traffic Sign Recognition Benchmark (GTSRB) dataset and achieves accurate recognition across 43 traffic sign categories.
Traffic sign recognition is a crucial component in autonomous driving systems. The goal of this project is to:
- Preprocess and augment traffic sign images.
- Train a CNN model from scratch.
- Compare performance with transfer learning (MobileNetV2 / VGG16).
- Evaluate, visualize, and deploy the final trained model.
Import all necessary libraries for deep learning, data preprocessing, and visualization.
Load the GTSRB dataset which contains images of 43 traffic sign classes.
Use ImageDataGenerator for data normalization and augmentation.
Design a CNN architecture suitable for traffic sign recognition.
Train the CNN on training data and validate on validation data.
Compute accuracy on the test set and visualize model performance.
Use MobileNetV2 or VGG16 for better accuracy through fine-tuning.
Save the trained CNN or transfer learning model for deployment.