|
|
--- |
|
|
license: mit |
|
|
tags: |
|
|
- image-classification |
|
|
- plant-disease |
|
|
- tensorflow |
|
|
- keras |
|
|
library_name: tensorflow |
|
|
--- |
|
|
|
|
|
# Leaf Disease Detection Model |
|
|
|
|
|
This model can detect diseases in plant leaves using EfficientNetB3 architecture. |
|
|
|
|
|
## Usage |
|
|
|
|
|
```python |
|
|
from huggingface_hub import hf_hub_download |
|
|
import tensorflow as tf |
|
|
import json |
|
|
|
|
|
# Download model and class indices |
|
|
model_path = hf_hub_download(repo_id="Rvish-glitch/leaf-disease-detection", filename="final_model.h5") |
|
|
indices_path = hf_hub_download(repo_id="Rvish-glitch/leaf-disease-detection", filename="class_indices.json") |
|
|
|
|
|
# Load model |
|
|
model = tf.keras.models.load_model(model_path) |
|
|
|
|
|
# Load class indices |
|
|
with open(indices_path, 'r') as f: |
|
|
class_indices = json.load(f) |
|
|
``` |
|
|
|
|
|
## Classes |
|
|
|
|
|
The model can detect various plant diseases across different crops including tomato, potato, apple, and corn. |
|
|
|
