Skip to content

πŸ–ΌοΈ CNN-based image classifier for CIFAR-10 using TensorFlow/Keras. Classifies 32x32 images into 10 categories with real-time predictions.

Notifications You must be signed in to change notification settings

Keval10github/Image-Recognition

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

15 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

KEVAL RAVANI

🌟 Image Recognition with CIFAR-10 Dataset 🌟

![CIFAR-10]

Image

πŸ“œ Overview

Welcome to the Image Recognition Project by Keval Ravani! πŸš€
This project uses a Convolutional Neural Network (CNN) to classify images from the CIFAR-10 dataset, which contains 60,000 32x32 color images across 10 classes, such as πŸ›©οΈ airplanes, πŸš— automobiles, 🐦 birds, 🐱 cats, and more.


πŸ“‚ Project Structure

The project follows a well-defined pipeline:

  1. Dataset Loading: CIFAR-10 dataset is loaded using TensorFlow.
  2. Preprocessing: Images are normalized for better model performance.
  3. Model Construction: A CNN is built using TensorFlow/Keras.
  4. Training: The model is trained on the dataset.
  5. Evaluation: The model's accuracy is tested on unseen data.
  6. Visualization: Random samples are visualized with predictions.

🎯 Features

  • πŸ” Image classification using CNNs
  • πŸ“Š Model training and evaluation
  • πŸ–ΌοΈ Visualization of sample predictions
  • 🏷️ Classes: Airplane, Automobile, Bird, Cat, Deer, Dog, Frog, Horse, Ship, Truck

πŸ“¦ Requirements

Dependencies

Ensure you have the following Python libraries installed:

  • tensorflow (compatible with Python 3.11)
  • matplotlib
  • random

Install them using:

Python Version

This project requires Python 3.11.x for compatibility with TensorFlow.


πŸ› οΈ Installation

  1. Clone the repository:
  2. Install dependencies:
  3. Run the script:

πŸ“– Code Walkthrough

1️⃣ Dataset Loading and Preprocessing

The CIFAR-10 dataset is loaded and split into training and testing sets: Images are normalized to scale pixel values between 0 and 1:

2️⃣ Visualization of Data Samples

Random samples from the training set are visualized with their labels:

3️⃣ CNN Model Architecture

The CNN consists of three convolutional layers followed by max-pooling layers and dense layers:

4️⃣ Training and Evaluation

The model is compiled and trained for two epochs: Evaluation on the test set:

5️⃣ Prediction and Visualization of Results

Random samples from the test set are visualized with their actual and predicted labels:


πŸ“Š Results

Metric Value
Training Accuracy ~59%
Test Accuracy ~60%

Sample Output Visualization:

Actual Label: 🐸 Frog

Predicted Label: 🐸 Frog

Image

πŸ“ˆ Future Enhancements

  • Train the model for more epochs to improve accuracy.
  • Experiment with data augmentation techniques.
  • Implement transfer learning using pre-trained models like ResNet or VGG.

πŸ‘¨β€πŸ’» Author

Keval Ravani
πŸ’Œ Email: kevalravani06@gmail.com
πŸ“… Date: April 13, 2025


πŸ“œ License

This project is licensed under the MIT License.


✨ Acknowledgments

Special thanks to the TensorFlow team for providing an easy-to-use framework and to the creators of the CIFAR-10 dataset.


πŸš€ Let's Collaborate!

Feel free to open issues or submit pull requests for improvements. Contributions are always welcome! 😊


About

πŸ–ΌοΈ CNN-based image classifier for CIFAR-10 using TensorFlow/Keras. Classifies 32x32 images into 10 categories with real-time predictions.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published