Skip to content

RezaeiAlireza/Image-Classifier-CNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 

Repository files navigation

Image Classifier using Convolutional Neural Networks (CNN)

This repository contains the implementation of an Image Classifier using Convolutional Neural Networks (CNN) built using Keras and TensorFlow. The project focuses on classifying images from the CIFAR-10 dataset.

Overview

The CIFAR-10 dataset consists of 60,000 32x32 color images in 10 classes, with 6,000 images per class. There are 50,000 training images and 10,000 test images. The dataset is divided into five training batches and one test batch, each with 10,000 images. The test batch contains exactly 1,000 randomly-selected images from each class. The training batches contain the remaining images in random order, but some training batches may contain more images from one class than another. Between them, the training batches contain exactly 5,000 images from each class.

Dependencies

  • Python 3.x
  • TensorFlow
  • Keras
  • NumPy
  • Matplotlib

Getting Started

  1. Clone this repository to your local machine.
  2. Install the required dependencies using pip: pip install -r requirements.txt
  3. Run the main script: python image_classifier_cnn.py

Project Structure

  • image_classifier_cnn.py: Main script for training and evaluating the CNN model.
  • requirements.txt: Contains the required dependencies for this project.

Model Architecture

The implemented CNN consists of several convolutional and pooling layers, followed by fully connected layers and a final softmax activation function for classification. The architecture was designed to achieve high accuracy while minimizing overfitting through the use of dropout layers.

Results

The model achieves competitive accuracy on the CIFAR-10 test dataset. Detailed results, including training and validation loss and accuracy, are displayed in the form of plots.

About

An Image Classifier using Convolutional Neural Networks (CNN) implemented in Keras to classify images from the CIFAR-10 dataset. The model consists of multiple convolutional, pooling, and dense layers for improved classification performance. The script includes visualization functions to display the images along with their predicted labels

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors