This project implements a Convolutional Neural Network (CNN) using PyTorch to classify images in the CIFAR-10 dataset into 10 object categories such as airplane, automobile, bird, cat, etc. The solution includes data preprocessing, model design, training, evaluation, and testing phases.
- Jupyter Notebook
- Python 3.9+
- PyTorch >= 2.0 (With Cuda support)
- torchvision
- matplotlib
- scikit-learn
- numpy
- tqdm
- seaborn
Install dependencies:
PyTorch https://pytorch.org/get-started/locally/
pip install torchvision matplotlib scikit-learn tqdm seaborn notebook