Downloads · 30 days
23
17% of all-time downloads
sudhir75/Harimitra
Harimitra is a image classification model from sudhir75. Use it when you need a label for an image. It is set up for keras. The card lists the license as mit.
Harimitra is a deep learning model for automated plant disease detection and classification. Built using Convolutional Neural Networks (CNN), this model can identify 38 different plant disease classes from leaf images…
Downloads · 30 days
23
17% of all-time downloads
All-time downloads
133
Public
Repo size
377 MB
Likes
2
Public
Click a slice to open those files.
.h594.2 MB · 74%
From the Hugging Face model README
Harimitra is a deep learning model for automated plant disease detection and classification. Built using Convolutional Neural Networks (CNN), this model can identify 38 different plant disease classes from leaf images, helping farmers and agricultural professionals diagnose plant health issues quickly and accurately.
This model was trained on the New Plant Diseases Dataset from Kaggle:
Each class represents a specific combination of plant type and disease condition (or healthy state). The dataset covers multiple plant species including tomato, potato, corn, grape, apple, and others, with various diseases such as bacterial spot, early blight, late blight, leaf mold, and healthy leaf conditions.
This model is designed to:
pip install -r requirement.txt
from tensorflow import keras
import numpy as np
from PIL import Image
# Load the model
model = keras.models.load_model('trained_model.keras')
# Load and preprocess image
img = Image.open('plant_leaf.jpg')
img = img.resize((256, 256)) # Dataset standard size
img_array = np.array(img) / 255.0 # Normalize to [0, 1]
img_array = np.expand_dims(img_array, axis=0)
# Make prediction
predictions = model.predict(img_array)
predicted_class = np.argmax(predictions, axis=1)
print(f"Predicted class: {predicted_class[0]}")
print(f"Confidence: {np.max(predictions)*100:.2f}%")
python main.py --image path/to/plant_leaf.jpg
The model was trained using the notebooks provided in this repository:
Train_plant_disease.ipynb - Complete training workflowTest_Plant_Disease.ipynb - Model evaluation and testingtraining_hist.json| File | Size | Description |
|---|---|---|
trained_model.keras | 31.4 MB | Model in Keras format (recommended) |
trained_model.h5 | 94.2 MB | Model in H5 format (legacy) |
main.py | 4.53 KB | Main inference script |
Train_plant_disease.ipynb | 552 KB | Training notebook |
Test_Plant_Disease.ipynb | 600 KB | Testing and evaluation notebook |
training_hist.json | 862 B | Training metrics and history |
requirement.txt | 126 B | Required dependencies |
home_page.jpeg | 75.9 KB | Application interface image |
Training metrics and performance details are available in training_hist.json. The model was evaluated on multiple metrics including:
The model can classify 38 different classes including various combinations of:
Plant Species: Apple, Blueberry, Cherry, Corn, Grape, Orange, Peach, Pepper, Potato, Raspberry, Soybean, Squash, Strawberry, Tomato
Disease Types: Bacterial spot, Early blight, Late blight, Leaf blight, Leaf scorch, Leaf mold, Septoria leaf spot, Spider mites, Target spot, Tomato Yellow Leaf Curl Virus, Tomato mosaic virus, Black rot, Esca, Cedar apple rust, Powdery mildew, and Healthy conditions
If you use this model in your research or application, please cite:
@misc{harimitra2025,
author = {sudhir75},
title = {Harimitra: Plant Disease Detection Model},
year = {2025},
publisher = {HuggingFace},
url = {https://huggingface.co/sudhir75/Harimitra}
}
@dataset{new_plant_diseases_dataset,
author = {vipoooool},
title = {New Plant Diseases Dataset},
year = {2020},
publisher = {Kaggle},
url = {https://www.kaggle.com/datasets/vipoooool/new-plant-diseases-dataset}
}
This model is released under the MIT License.
For questions, issues, or contributions:
training_hist.json for model performance metrics