/FloraVision

In this Project we classify different categories of Flowers using a Pre-Trained model and also using the functionality of Streamlit Application to build an API or frontend for the Project along with the various libraries of python for proper functioning of Model.

Primary LanguagePythonMIT LicenseMIT

GitHub license GitHub contributors GitHub issues GitHub pull-requests PRs Welcome

GitHub watchers GitHub forks GitHub stars

Open in Visual Studio Code

FloraVision 🌷:

Flower-Image-Classification-Streamlit-TensorFlow Designer (1)

A basic web-app for image classification using Streamlit and TensorFlow.

It classifies the given image of a flower into one of the following five categories :-

  1. Daisy
  2. Dandelion
  3. Rose
  4. Sunflower
  5. Tulip

Notes:

  • A simple flower classification model was trained using TensorFlow.
  • The weights are stored as flower_model_trained.hdf5.
  • The code to train the modify and train the model can be found in model.py.
  • The web-app created using Streamlit can be found in app.py

Commands ✍️:

To run the app locally, use the following command :-
streamlit run app.py

The webpage should open in the browser automatically.
If it doesn't, the local URL would be output in the terminal, just copy it and open it in the browser manually.
By default, it would be http://localhost:8501/

Click on Browse files and choose an image from your computer to upload.
Once uploaded, the model will perform inference and the output will be displayed.

Output 📁:

For more output images visit: Link out_1

Reference 🧧:

Requirements💻 :

Ensure you have the following dependencies installed:

  • Python (version 3.9.x || 3.12.x)
  • IDE: VS-CODE or collab
  • Virtual-environment(venv)
  • Other dependencies (refer to the requirements.txt)

You can install the required Python packages using:

pip install -r requirements.txt

Setup 💿:

  • Clone the repository:
git clone https://github.com/SINGHxTUSHAR/FloraVision.git
cd FloraVision
  • Create a virtual environment (optional but recommended):
python -m venv venv
  • Activate the virtual environment:
    • On Windows:
    venv\Scripts\activate
    • On macOS/Linux:
    source venv/bin/activate

Contributing 📌:

If you'd like to contribute to this project, please follow the standard GitHub fork and pull request process. Contributions, issues, and feature requests are welcome!

Suggestion 🚀:

If you have any suggestions for me related to this project, feel free to contact me at tusharsinghrawat.delhi@gmail.com or LinkedIn.

License 📝:

This project is licensed under the MIT License - see the LICENSE file for details.