HW3

CIFAR-10 Image Classification with CNN (PyTorch)

Project Description

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.

Download jpnb file and run using notebook!

Environment Setup

Requirements

• 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