catalinalopera / ObjectClassification-

This project demonstrates how to build and train a Convolutional Neural Network (CNN) using TensorFlow and Keras to classify images as pizza or not pizza. The model uses image augmentation techniques to improve generalization and prevent overfitting, and is suitable for beginner-level computer vision tasks.

Repository from Github https://github.comcatalinalopera/ObjectClassification-Repository from Github https://github.comcatalinalopera/ObjectClassification-

πŸ• Pizza vs No Pizza - Image Classification with CNN

πŸ“Œ About

This project demonstrates how to build and train a Convolutional Neural Network (CNN) using TensorFlow and Keras to classify images as pizza or not pizza. It applies basic image augmentation techniques to improve generalization and prevent overfitting. This is an ideal beginner-friendly computer vision project.


πŸš€ Features

  • Binary image classification: pizza vs no pizza
  • Image augmentation using ImageDataGenerator
  • CNN model with dropout to reduce overfitting
  • Live visualization of training accuracy and loss
  • Easily extendable to other binary classification problems

🧰 Technologies Used

  • Python 3.x
  • TensorFlow / Keras
  • NumPy
  • Matplotlib
  • Google Colab / Jupyter Notebook / VSCode / Kaggle

πŸ—‚οΈ Dataset Structure

The dataset should be organized as follows:

data/ train/ pizza/ no_pizza/ test/ pizza/ no_pizza/

About

This project demonstrates how to build and train a Convolutional Neural Network (CNN) using TensorFlow and Keras to classify images as pizza or not pizza. The model uses image augmentation techniques to improve generalization and prevent overfitting, and is suitable for beginner-level computer vision tasks.


Languages

Language:Jupyter Notebook 100.0%