CIFAR-10 Image Classification using PyTorch
A Convolutional Neural Network (CNN) built with PyTorch to classify images from the CIFAR-10 dataset.
Data → Preprocessing → CNN → Training → Evaluation → Prediction
CIFAR-10 contains 60,000 color images across 10 classes:
✈️ Airplane · 🚗 Automobile · 🐦 Bird · 🐱 Cat · 🦌 Deer · 🐶 Dog · 🐸 Frog · 🐴 Horse · 🚢 Ship · 🚚 Truck
Input Image
↓
Convolution
↓
Activation + Pooling
↓
Feature Extraction
↓
Fully Connected Layers
↓
10-Class Prediction
| Metric | Result |
|---|---|
| Dataset | CIFAR-10 |
| Framework | PyTorch |
| Model | CNN |
| Test Accuracy | 75.07% |
| Classes | 10 |
Python · PyTorch · TorchVision · Matplotlib · Google Colab
git clone https://github.com/sumitjhadev/cnn-image-classification.git
cd cnn-image-classificationOpen CNN.ipynb in Google Colab or Jupyter Notebook and run the cells.
cnn-image-classification/
└── CNN.ipynb
Sumit Jha
⭐ If you found this project useful, consider starring the repository!
Built with 🧠 PyTorch & ❤️ for Deep Learning
```