Skip to content

This project demonstrates image classification using a Convolutional Neural Network (CNN) on the CIFAR-10 dataset. The model is trained to classify images into one of 10 classes.

Notifications You must be signed in to change notification settings

SreeEswaran/Image-classification-using-CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

11 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Image Classification using CNN

This project demonstrates image classification using a Convolutional Neural Network (CNN) on the CIFAR-10 dataset. The model is trained to classify images into one of 10 classes.

Features

  • Load Data: Load and preprocess CIFAR-10 image data.
  • Train Model: Train a CNN model on the CIFAR-10 training dataset.
  • Evaluate Model: Evaluate the trained model on the CIFAR-10 test dataset.
  • Jupyter Notebook: Interactive notebook for visualizing the image classification process.

Download Data

Download the CIFAR-10 dataset from the official website and extract it into the data folder. Alternatively, you can run the following script to automatically download and extract the dataset:

mkdir -p data
cd data
wget https://www.cs.toronto.edu/~kriz/cifar-10-python.tar.gz
tar -xvzf cifar-10-python.tar.gz
cd ..

Setup

  1. Clone the repository and install dependencies:
git clone https://github.com/SreeEswaran/Image-classification-using-CNN.git
cd Image-classification-using-CNN
  1. Install the depedencies
    pip install -r requirements.txt

Usage

  1. Train the model
    python scripts/train.py
  2. Evaluate the model
    python scripts/evaluate.py

Interactive notebook

Open the Jupyter notebook:

jupyter notebook notebooks/Image_classification_using_CNN.ipynb

About

This project demonstrates image classification using a Convolutional Neural Network (CNN) on the CIFAR-10 dataset. The model is trained to classify images into one of 10 classes.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published