Skip to content

This project demonstrates image classification using a Convolutional Neural Network (CNN) built with TensorFlow and Keras. It includes data preprocessing, model training, evaluation, and visualization to classify images into their respective categories.

Notifications You must be signed in to change notification settings

Visshu78/image-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image Classification using Convolutional Neural Networks (CNN)

This project demonstrates how to build an image classification model using a Convolutional Neural Network (CNN) with TensorFlow and Keras. It classifies images into categories based on features learned from the data.


Project Files

  • Image_classification.ipynb – Main Jupyter notebook with full code.
  • requirements.txt – Python libraries used in the project.
  • README.md – Project overview and setup instructions.

What the Project Does

  • Loads and processes image data using ImageDataGenerator.
  • Builds a CNN with layers like Conv2D, MaxPooling2D, Dropout, and Dense.
  • Trains the model and plots training vs. validation accuracy and loss.
  • Tests the trained model on new images for prediction.

Technologies Used

  • Language: Python
  • Libraries: TensorFlow, Keras, NumPy, Matplotlib
  • Model Architecture: Convolutional Neural Network (CNN)

🔧 Setup Instructions

Clone the Repository

git clone https://github.com/your-username/image-classification-cnn.git
cd image-classification-cnn

About

This project demonstrates image classification using a Convolutional Neural Network (CNN) built with TensorFlow and Keras. It includes data preprocessing, model training, evaluation, and visualization to classify images into their respective categories.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published