This project explores image classification on the CIFAR-10 dataset using two approaches:
- A Custom CNN built from scratch
- Transfer Learning with pre-trained models
We compare performance, training times, and trade-offs between the two approaches.
ProjectCifar.ipynb
β Main notebook with all experimentsREADME.md
β Project documentation (this file)
- CIFAR-10:
- 60,000 color images (32Γ32 pixels)
- 10 categories: airplane, automobile, bird, cat, deer, dog, frog, horse, ship, truck
- Train set: 50,000 images, Test set: 10,000 images
-
Custom CNN
- Several convolutional + pooling layers
- Flatten β Dense β Output
- Trained from scratch
-
Transfer Learning
- Used pre-trained models (e.g., MobileNet with different training layers)
- Fine-tuned on CIFAR-10 dataset
- Faster convergence and improved accuracy
Install dependencies with:
pip install -r requirements.txt