Skip to content

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

🌿 Plant Disease Detection & Severity Estimation

Python TensorFlow YOLOv8 Streamlit

An intelligent plant disease detection pipeline using YOLOv8 for leaf detection, MobileNetV2 for disease classification, and YOLOv8-seg for accurate severity estimation with spray recommendations.


🔄 Pipeline Architecture

Pipeline Architecture


🎯 Severity Estimation

Heuristic vs Segmentation Model

Severity Comparison

Visual Example


✨ Features

Feature Description
🔍 Leaf Detection YOLOv8 detects individual leaves
🏷️ Disease Classification 15 disease classes (Tomato, Potato, Pepper)
🔬 Segmentation Severity YOLOv8-seg for accurate (~85-95%) severity
📊 Heuristic Fallback Color-based analysis if no seg model
🎯 Spray Tiers NO_ACTION → LOW → MEDIUM → HIGH

🚀 Quick Start

# Install
pip install -r requirements.txt

# Run Web UI
streamlit run scripts/streamlit_app.py

Models Required:

Model Path Purpose
YOLOv8 Detection models/yolov8_leaf.pt Leaf detection
MobileNetV2 models/mobilenetv2_disease.keras Disease classification
YOLOv8 Segmentation models/yolov8_seg.pt Severity estimation ⭐

🎚️ Settings Guide

Setting Value Use Case
YOLO Confidence 0.35 General use ✅
Segmentation Toggle ON More accurate severity

👨‍💻 Author

Anuj Tripathi | EDP Project


About

Plant Disease Detection & Severity Estimation

Topics

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages