Multiclass Image Classification of 7 different classes using transfer learning with pretrained models.
Kaggle Dataset Link - Multi-Class Image Classification Dataset
This Dataset has images of 7 different classes -
- Bus
- Car
- Cat
- Cricket
- Dog
- Football
- Product
I've used 4 different CNN models to train these images -
- CNN model bulit from scratch
- Hypertuned CNN model bulit from scratch
- Pretrained model VGG-16
- Pretrained model Inception-V3
I've used Transfer Learning to train these pretrained models, VGG-16 and Inception-V3, on a custom dataset. These 2 models have the best accuracy and f1-score.