Skip to content

A Streamlit-based web application for vegetable classification using a deep learning model. Developed as part of a university project (This project is submitted to Dr. Ahmed Badawy and Eng. Noor Eldeen Magdy, Faculty of Engineering, Helwan University, as part of a coursework requirement).

License

Notifications You must be signed in to change notification settings

imnotamr/VegetableClassificationAppFEHU

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

49 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Vegetable Classification App

This project features a Vegetable Classification App, built using a state-of-the-art Convolutional Neural Network (CNN). The app allows users to upload images of vegetables and receive accurate classifications, along with confidence scores for each category. Check out the App ---> Vegetable Classification App

Features

Image Upload:

Upload vegetable images (JPG, PNG, or JPEG formats).

Deep Learning Model:

Utilizes a CNN trained on 15 vegetable classes for high accuracy.

Confidence Scores:

Visualizes classification probabilities with a bar chart.

Streamlit Deployment:

Easy-to-use interface accessible via a web browser.

📂 Project Structure

Vegetable-Classification-App/

├── streamlit_app.py

├── requirements.txt
├── model/

│ └── Vegetable_model_last.h5
├── README.md

└── assets/

How It Works

Upload an Image:

Users upload a vegetable image in JPG/PNG format.

Model Prediction:

The app uses the pre-trained CNN to classify the vegetable.

Display Results:

Predicted vegetable name and Confidence scores for all 15 classes, displayed as a bar chart.

Supported Vegetable Classes:

Bean, Bitter_Gourd, Bottle_Gourd, Brinjal, Broccoli, Cabbage, Capsicum, Carrot, Cauliflower, Cucumber, Papaya, Potato, Pumpkin, Radish, Tomato

Deployment

Using Streamlit Cloud The app is deployed via Streamlit Cloud for easy access. Check it out here: Vegetable Classification App

Model Details

Framework:

TensorFlow/Keras

Model Type:

Convolutional Neural Network (CNN)

Classes:

15 vegetable types

Training Dataset:

High-resolution vegetable images

Output Layer:

Softmax for multi-class classification

Contributing

Contributions are welcome! If you’d like to improve the model, app interface, or documentation:

  1. Fork the repository.
  2. Create a feature branch.
  3. Submit a pull request.

License

This project is licensed under the MIT License. Feel free to use and modify the code as needed.

Acknowledgments

Team Members:

Amr Ahmed, Mohamed Yasser, Omar Khaled, Ibrahim Mahmoud.

Frameworks:

TensorFlow, Keras, Streamlit.

Special Thanks:

Open-source communities for making this possible.

Contact

If you have any questions or suggestions, feel free to reach out:

GitHub: @imnotamr

🌟 Don’t forget to star the repository if you find it useful!

About

A Streamlit-based web application for vegetable classification using a deep learning model. Developed as part of a university project (This project is submitted to Dr. Ahmed Badawy and Eng. Noor Eldeen Magdy, Faculty of Engineering, Helwan University, as part of a coursework requirement).

Topics

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages