Skip to content

This project aims to classify different types of fruits using deep learning. The objective is to build a model that can accurately identify the type of fruit based on images.

Notifications You must be signed in to change notification settings

SiddiquiZainab/FruitC

Repository files navigation

Fruit Classification using ML Models

Project Overview

This project aims to classify different types of fruits using machine learning models. The objective is to build a model that can accurately identify the type of fruit based on images.

Table of Contents

Data

The dataset used for this project consists of images of different types of fruits. Each image is labeled with the type of fruit it represents. The dataset includes a variety of fruits such as apples, bananas, oranges, and more. There are a total of 10 categories in the target.

Sample Images of class: apple image Sample images of class: avocado image Sample images of class: banana image Sample images of class: cherry image Sample images of class: kiwi image Sample images of class: mango image Sample images of class: orange image Sample images of class: pineapple image Sample images of class: strawberries image Sample images of class: watermelon image

Link to dataset: https://www.kaggle.com/datasets/karimabdulnabi/fruit-classification10-class

Methodology

The project implements Transfer Learning using pre-trained CNN of MobileNetV2 with ImageNet weights. No image preprocessing has been done. At the time of creation of data generators, the images are resized to an input shape of (224, 224, 3). The images are RGB (colored).

The architecture is as follows:

image

Results

The model attained a training accuracy of 97.70% and a testing accuracy of 67.22% in 23 epochs. These results demonstrate the model's ability to effectively classify different types of fruits based on images.

The model predictions are as follows:

image

Applications

This fruit classification model has several potential applications:

  • Automated Sorting Systems: Can be used in agricultural and food industries for automated sorting of fruits.
  • Retail: Helps in automating the identification and pricing of fruits in grocery stores.
  • Dietary Tracking: Assists in dietary tracking apps to automatically identify and log fruits.

Contact

For any questions or suggestions, please contact on LinkedIn: https://www.linkedin.com/in/siddiquizainab/

About

This project aims to classify different types of fruits using deep learning. The objective is to build a model that can accurately identify the type of fruit based on images.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published