Skip to content

This project focuses on the development of an intelligent and user-friendly web application that can accurately detect medicinal plants using deep learning techniques. By leveraging the power of YOLOv8 (You Only Look Once) object detection model and Streamlit, this application allows users to upload images of plants and receive real-time prediction

Notifications You must be signed in to change notification settings

Rohith1202/Medical-Plant-Detection-using-YOLO-V8-and-Generative-AI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

93 Commits
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

🌿 Medical Plant Detection using YOLOv8 and Generative AI

🧠 Project Overview

This project focuses on the development of an intelligent and user-friendly web application that can accurately detect medicinal plants using deep learning techniques. By leveraging the power of YOLOv8 (You Only Look Once) object detection model and Streamlit, this application allows users to upload images of plants and receive real-time predictions on the identified medicinal species.

This tool is designed to support botanical research, herbal medicine identification, and educational purposes, especially in rural and tribal areas where access to professional botanical expertise may be limited.


🎯 Objectives

  • Build a lightweight and accurate object detection model capable of recognizing multiple medicinal plants.
  • Deploy an interactive web interface using Streamlit for ease of use.
  • Enable image upload, display the most confident plant classification, and provide downloadable reports.
  • Maintain user-specific logs with timestamped detection data (name, age, and results).

πŸ› οΈ Technology Stack

Component Description
Model YOLOv8 (Ultralytics) pre-trained + custom
Framework PyTorch, Streamlit
Language Python
Interface Streamlit-based UI
Tools & Libraries OpenCV, NumPy, Pandas, PIL, os, datetime

πŸ” Model Training & Optimization

  • The YOLOv8 model was trained using a custom dataset consisting of labeled medicinal plant images.
  • Training involves:
    • Data preprocessing and augmentation
    • Bounding box annotations in YOLO format
    • Model fine-tuning and validation
  • The final model is exported as best.pt, ready for inference.

πŸ“· Streamlit Application Features

  • 🌱 Upload an image for medicinal plant detection.
  • πŸ”Ž Detect and show only the most confident plant name from the image.
  • 🧾 Display the result clearly below the image.
  • πŸ§‘β€πŸ’Ό Collect user name and age for logging purposes.
  • ⏰ Show a real-time clock on the interface.
  • πŸ“₯ Download detection results and image as a .txt log file.
  • ✨ Custom UI with background styling for an enhanced experience.

πŸ“ File Structure

β”œβ”€β”€ Medical Plant Detection.ipynb   # Jupyter Notebook for model loading and Streamlit app
β”œβ”€β”€ best.pt                         # Trained YOLOv8 model
β”œβ”€β”€ plant_images/                   # Folder with test or sample plant images
β”œβ”€β”€ requirements.txt                # Required Python packages
β”œβ”€β”€ utils/                          # Helper functions (optional)
└── README.md                       # Project documentation

πŸ§ͺ Sample Use Case

  1. Run the Streamlit app:

    streamlit run app.py
  2. Upload a plant image.

  3. Fill in your name and age in the provided fields.

  4. View the detection result and download the output file.


πŸ“₯ Sample Output File (TXT)

Name: Rohith
Age: 22
Timestamp: 2025-05-29 17:05:23
Detected Plant: Ocimum tenuiflorum (Tulsi)

πŸš€ Getting Started

Prerequisites

  • Python 3.8+
  • YOLOv8 (ultralytics package)
  • Streamlit
  • OpenCV, Pillow, Pandas

Install dependencies:

pip install -r requirements.txt

Dataset

This project utilizes a comprehensive medical plant detection dataset hosted on Roboflow Universe. The dataset contains high-quality annotated images of various medicinal plants, specifically curated for object detection and classification tasks.

Dataset URL: https://universe.roboflow.com/pkm-cj9yb/plant-detection-iqmnt

Dataset Features

  • High-resolution images of medicinal plants
  • Accurate bounding box annotations
  • Multiple export formats (YOLO, COCO, Pascal VOC)
  • Pre-processing and augmentation options
  • Suitable for training custom plant detection models

Usage

The dataset is used to train our deep learning models for accurate plant identification and classification, enabling the system to recognize various medicinal plants with high precision.

Run the Application

streamlit run Medical_Plant_Detection.py

πŸŽ“ Applications

  • Medicinal plant identification in remote areas.
  • Assisting researchers and herbal practitioners.
  • Building botanical datasets for AI.
  • Educational tools for plant taxonomy.

Screenshots

Authentication

image

User login interface for accessing the medical plant detection system

Plant Detection

image

Upload image interface for plant identification

image

Real-time plant detection using webcam

AI Assistant

image

Interactive AI chatbot for plant-related queries

Detection History

image

View previous plant detection results and analysis

Additional Features

User Feedback

image

User feedback interface for system improvement

About Section

image

Information about the medical plant detection system



πŸ‘¨β€πŸ’» About Me

Hi, I’m Rohith Boppana β€” a passionate and driven final-year B.Tech student in Computer Science and Engineering with a specialization in Artificial Intelligence & Machine Learning.

I'm deeply interested in building real-world tech solutions that combine data, intelligence, and intuitive design. My academic journey and hands-on projects reflect a strong foundation in both theory and practical application.

πŸ‘‡ My Core Interests

  • πŸ€– Artificial Intelligence & Machine Learning
  • πŸ” Data Science & Analytics
  • πŸ“Š BI Dashboards & Predictive Modeling
  • πŸ’‘ Problem-Solving with Scalable Technologies

I enjoy translating business needs and data insights into impactful software solutions that solve real problems and enhance user experiences.


πŸ”— Let’s Connect

πŸ“« LinkedIn
Let’s connect and grow professionally:
linkedin.com/in/rohith-boppana-39ab57279

🌐 Portfolio
Explore my latest work, skills, and projects here:
rohith-boppana.vercel.app


πŸ’‘ β€œFinal-year student, forever learner β€” building the future, one project at a time.”

Feel free to explore my repositories and reach out for collaborations, internships, or to discuss innovative ideas!

About

This project focuses on the development of an intelligent and user-friendly web application that can accurately detect medicinal plants using deep learning techniques. By leveraging the power of YOLOv8 (You Only Look Once) object detection model and Streamlit, this application allows users to upload images of plants and receive real-time prediction

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published