Skip to content

README.md

Latest commit

 

History

History
82 lines (67 loc) · 2.08 KB

README.md

File metadata and controls

82 lines (67 loc) · 2.08 KB

Image Classification using ANN & CNN

📸 Project Overview

This project focuses on image classification using the CIFAR-10 dataset. Two models, ANN and CNN, are trained on 32x32 color images spanning 10 different classes. The dataset undergoes preprocessing, after which the models are built, trained, and evaluated.

🚀 Features

  • 🔹 Artificial Neural Network (ANN) implementation
  • 🔹 Convolutional Neural Network (CNN) implementation
  • 🔹 Utilizes TensorFlow and Keras
  • 🔹 CIFAR-10 dataset for multi-class classification
  • 🔹 Model evaluation & visualization with Matplotlib

📂 Dataset

The CIFAR-10 dataset consists of 60,000 images (50,000 for training, 10,000 for testing) classified into: 🔹 Airplane ✈️
🔹 Automobile 🚗
🔹 Bird 🐦
🔹 Cat 🐱
🔹 Deer 🦌
🔹 Dog 🐶
🔹 Frog 🐸
🔹 Horse 🐴
🔹 Ship 🚢
🔹 Truck 🚛

🖥 GUI Preview

An interactive GUI is included for testing the model with custom images.

🔹 Features

  • Upload an image for classification
  • Get real-time predictions
  • Compare ANN vs CNN performance

GUI Preview

📌 Installation & Usage

🔹 Prerequisites

Ensure you have Python installed along with required libraries:

pip install tensorflow numpy matplotlib tkinter

🔹 Clone the Repository

git clone https://github.com/yourusername/image-classification.git
cd image-classification

🔹 Run the Training Script

python train_model.py

🔹 Launch the GUI

python gui.py

📊 Model Performance

The models are trained and evaluated based on accuracy and loss metrics.

Model Accuracy (Test)
ANN 65%
CNN 85%

🤝 Contributing

Feel free to contribute! Open an issue or submit a pull request.

📜 License

This project is licensed under the MIT License.

📞 Contact

For any queries, reach out to me at: your.email@example.com