Data : https://www.kaggle.com/datasets/ahemateja19bec1025/traffic-sign-dataset-classification
Pre-trained Models used:
- VGG16
- Xception
- ResNet50
- Loaded the traffic sign images from the directory structure.
- Used ImageDataGenerator to rescale the images and split the data into training and validation sets.
- Loaded the pre-trained VGG16 model without the top layer.
- Added a custom dense layer for classification.
- Compiled and trained the model.
- Saved the model and evaluated its performance, achieving high accuracy.
- Loaded the pre-trained Xception model.
- Added custom layers for classification.
- Compiled and trained the model with early stopping.
- Evaluated the model, achieving reasonable accuracy but lower than VGG16.
- Loaded the pre-trained ResNet50 model.
- Added custom layers for classification.
- Compiled the model and trained the model.
- Evaluated the model, achieving reasonable accuracy but lower than VGG16.