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.
- 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
- Python 3.x
- TensorFlow / Keras
- NumPy
- Matplotlib
- Google Colab / Jupyter Notebook / VSCode / Kaggle
The dataset should be organized as follows:
data/ train/ pizza/ no_pizza/ test/ pizza/ no_pizza/