An open API service indexing awesome lists of open source software.

https://github.com/bhaveshbhakta/vegetable-classification-using-inceptionv3

Vegetable Classification
https://github.com/bhaveshbhakta/vegetable-classification-using-inceptionv3

cnn cnn-classification deep-learning inceptionv3 vegetable-classification

Last synced: 3 months ago
JSON representation

Vegetable Classification

Awesome Lists containing this project

README

          

# Vegetable Classification Using InceptionV3

## Overview
This project focuses on classifying different types of vegetables using the **InceptionV3** deep learning model. The dataset contains images of various vegetables, and the model is trained to recognize and categorize them accurately. The workflow includes data preprocessing, augmentation, model training, evaluation, and visualization of results.

## Technical Highlights
- **Dataset:** The dataset is sourced from Kaggle and contains labeled images of 15 vegetable categories.
- **Preprocessing:** Includes image resizing, augmentation (rotation, zoom, brightness adjustments), and normalization.
- **Model:** Uses **InceptionV3** as a pre-trained model with transfer learning, adding fully connected layers for classification.
- **Training:** The model is trained using categorical cross-entropy loss and the Adam optimizer with early stopping and checkpointing.
- **Evaluation:** Metrics like accuracy, loss curves, confusion matrix, and classification reports are used for performance analysis.

## Purpose
The primary goal of this project is to develop an efficient deep-learning model for vegetable classification. This helps in automating the process of identifying vegetables from images, reducing human effort and potential errors. The project also explores the effectiveness of transfer learning in image classification tasks.

## Applications
- **Smart Retail & Inventory Management:** Automating vegetable recognition in grocery stores for efficient checkout and stock monitoring.
- **Agriculture & Supply Chain:** Assisting farmers and suppliers in sorting and classifying vegetables.
- **Food Industry:** Enhancing food processing automation by categorizing vegetables accurately.
- **Educational Use:** Providing a practical deep learning example for image classification using TensorFlow and Keras.

**Installation**
To set up and run the project, follow these steps:
1. **Clone the repository**:
```bash
git clone https://github.com/BhaveshBhakta/Vegetable-Classification-Using-InceptionV3.git
```
2. **Navigate to the project directory**:
```bash
cd Vegetable-Classification-Using-InceptionV3
```
3. **Run the jupyter notebook**:

**Collaboration**
Contributions are welcome! Feel free to fork the repository, make improvements, and submit a pull request.