Skip to content

This project explores image classification on the CIFAR-10 dataset using two approaches: 1. A Custom CNN built from scratch 2. Transfer Learning with pre-trained models

License

Notifications You must be signed in to change notification settings

Anirudh-Unni/CIFAR-ImageClassification-CNN

Repository files navigation

🧠 Image Classification with CNN and Transfer Learning

This project explores image classification on the CIFAR-10 dataset using two approaches:

  1. A Custom CNN built from scratch
  2. Transfer Learning with pre-trained models

We compare performance, training times, and trade-offs between the two approaches.


πŸ“‚ Project Structure

  • ProjectCifar.ipynb β†’ Main notebook with all experiments
  • README.md β†’ Project documentation (this file)

πŸ“Š Dataset

  • 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

βš™οΈ Methods

  1. Custom CNN

    • Several convolutional + pooling layers
    • Flatten β†’ Dense β†’ Output
    • Trained from scratch
  2. Transfer Learning

    • Used pre-trained models (e.g., MobileNet with different training layers)
    • Fine-tuned on CIFAR-10 dataset
    • Faster convergence and improved accuracy

πŸ“¦ Requirements

Install dependencies with:

pip install -r requirements.txt

About

This project explores image classification on the CIFAR-10 dataset using two approaches: 1. A Custom CNN built from scratch 2. Transfer Learning with pre-trained models

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published