Fruit Classification Model

Overview

This project uses a Convolutional Neural Network built with TensorFlow and Keras to classify images of fruits. The classifier can distinguish between six different categories: fresh apples, fresh bananas, fresh oranges, rotten apples, rotten bananas, and rotten oranges.

Install Dependencies

pip install tensorflow keras numpy pillow

Dataset Preparation

1. Download dataset from Kaggle

2. Extract dataset in project folder

3. Remove duplicate dataset

rm -r PyVision/original_data_set/original_data_set

Training the model

python train.py

Make Predictions

python predict.py images/banana.jpg