Skip to content

Image classification is normally done using CNN, but i ahve done using ANN. learning concepts one by one

Notifications You must be signed in to change notification settings

NomNomNarang/Image-Classification-ANN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

ย 

History

6 Commits
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 
ย 

Repository files navigation

๐Ÿง  Classification using Artificial Neural Network (ANN)

๐Ÿ“˜ Overview

This project demonstrates image classification using an Artificial Neural Network (ANN) built with TensorFlow and Keras.
The dataset was imported directly from keras.datasets. The goal was to understand how neural networks learn and to observe the difference in training speed between CPU and GPU execution.


๐Ÿš€ Project Highlights

  • Implemented a basic ANN model for image classification.
  • Used TensorFlow/Keras for building, training, and evaluating the model.
  • Learned how model training speed varies between CPU and GPU.
  • Observed that deep learning models train significantly faster on GPUs compared to CPUs.

๐Ÿงฉ Tech Stack

  • Language: Python
  • Libraries: TensorFlow, Keras, NumPy, Matplotlib
  • Dataset: Imported from keras.datasets (e.g., MNIST or CIFAR-10)

โš™๏ธ Installation & Setup

  1. Clone this repository:
    git clone https://github.com/your-username/image-classification-ann.git
    cd image-classification-ann
    
    ๐Ÿ–ฅ๏ธ Performance Note
    

Training the model on a CPU (Intel integrated graphics) was slow, demonstrating how much time deep learning models require without GPU acceleration.

๐Ÿ’ก Pro Tip: For deep learning projects, prefer a laptop or system with an NVIDIA GPU that supports CUDA and cuDNN. It makes model training 10โ€“50ร— faster.

Results

Model successfully classified images from the dataset. Achieved reasonable accuracy after several epochs. Observed significant performance improvement potential with GPU usage.

About

Image classification is normally done using CNN, but i ahve done using ANN. learning concepts one by one

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published