--- library_name: keras license: apache-2.0 tags: - potato-disease-classification - plant-disease - computer-vision - agriculture - densenet169 - image-classification datasets: - mendeley model-index: - name: densenet169_mendeley results: - task: type: image-classification metrics: - type: accuracy value: 0.9156 - type: f1-score value: 0.9162 - type: precision value: 0.9198 - type: recall value: 0.9156 language: - en --- # DenseNet169 Potato Leaf Disease Classifier (Mendeley Dataset) ## Model Description This model is a fine-tuned **DenseNet169** CNN trained for potato leaf disease classification on the **Mendeley Potato Disease Dataset**. The model is designed to detect and classify 7 different potato plant disease categories using leaf images. ### Key Features - **Architecture**: DenseNet169 (pre-trained on ImageNet, fine-tuned on Mendeley dataset) - **Framework**: TensorFlow/Keras - **Input**: RGB images (224×224 pixels) - **Classes**: 7 disease categories - **Format**: `.keras` (Keras 3 format) ## Model Performance ### Overall Metrics - **Accuracy**: 91.56% - **Precision**: 91.98% - **Recall**: 91.58% - **F1-Score**: 91.62% - **Matthews Correlation Coefficient (MCC)**: 0.8976 - **Balanced Accuracy**: 90.47% ### Per-Class Performance | Class | Precision | Recall | F1-Score | Support | |-------|-----------|--------|----------|---------| | Bacteria | 0.97 | 0.98 | 0.98 | 114 | | Fungi | 0.90 | 0.92 | 0.91 | 150 | | Healthy | 0.75 | 0.97 | 0.85 | 40 | | Nematode | 0.92 | 0.79 | 0.85 | 14 | | Pest | 0.88 | 0.87 | 0.88 | 122 | | Phytophthora | 0.94 | 0.91 | 0.93 | 69 | | Virus | 0.98 | 0.89 | 0.93 | 107 | **Test Set Size**: 616 images ## Dataset Information ### Mendeley Potato Disease Dataset - **Total Images**: ~3,076 images - **Classes**: 7 disease categories - Bacteria - Fungi - Healthy - Nematode - Pest - Phytophthora - Virus - **Data Source**: [Mendeley Data - Potato Leaf Diseases (ptz377bwb8)](https://data.mendeley.com/datasets/ptz377bwb8/1) - **Characteristics**: Real-world field images with varying lighting, angles, and natural conditions ## Usage ### Installation ```bash pip install tensorflow keras ``` ### Load and Use the Model ```python import tensorflow as tf from tensorflow import keras from PIL import Image import numpy as np # Load the model model = keras.models.load_model('best.keras') # Prepare image image = Image.open('path/to/potato_leaf.jpg') image = image.resize((224, 224)) image_array = np.array(image) / 255.0 image_array = np.expand_dims(image_array, axis=0) # Make prediction predictions = model.predict(image_array) class_names = ['Bacteria', 'Fungi', 'Healthy', 'Nematode', 'Pest', 'Phytophthora', 'Virus'] predicted_class = class_names[np.argmax(predictions)] confidence = np.max(predictions) print(f"Predicted Disease: {predicted_class}") print(f"Confidence: {confidence:.2%}") ``` ### Using Hugging Face Transformers ```python # Coming soon - optimized inference pipeline ``` ## Training Details ### Architecture - **Base Model**: DenseNet169 (ImageNet pre-trained weights) - **Input Size**: 224×224×3 - **Total Parameters**: ~14M - **Custom Head**: Fine-tuned on Mendeley dataset ### Data Preprocessing - Image resizing: 224×224 - Normalization: ImageNet normalization - Augmentation: Random rotations, flips, brightness adjustments ### Training Configuration - **Optimizer**: Adam - **Loss Function**: Categorical Cross-Entropy - **Metrics**: Accuracy, Precision, Recall, F1-Score - **Early Stopping**: Enabled to prevent overfitting ## Model Card Information - **Model Name**: DenseNet169 Potato Leaf Disease Classifier - **Created**: 2026 - **Framework**: Keras/TensorFlow - **Model Type**: Image Classification (CNN) - **License**: Apache 2.0 ## Intended Use This model is intended for: - **Primary Use**: Classification of potato leaf diseases in agricultural applications - **Supported Use Cases**: - Automated disease detection systems - Agricultural advisory systems - Precision farming applications - Research and baseline comparisons ## Limitations - **Dataset Scope**: Trained exclusively on Mendeley dataset; performance may vary on other potato disease datasets - **Class Imbalance**: Some classes (Nematode: 14 samples) have limited training data - **Environmental Factors**: Model trained on field conditions; controlled environment images may perform differently - **Image Quality**: Requires clear leaf images; heavily obscured or blurry images may reduce accuracy - **Biological Variation**: Limited to potato plant diseases; not suitable for other crop diseases ## Related Models - **DenseNet201 Mendeley**: Alternative architecture with similar performance - **ResNet50 Mendeley**: Baseline comparison model - **Models on PlantVillage Dataset**: 3-class variant (Early Blight, Late Blight, Healthy) ## Citation If you use this model, please cite: ```bibtex @dataset{mendeley_potato_disease, title = {Potato Leaf Diseases Dataset}, url = {https://data.mendeley.com/datasets/ptz377bwb8/1} } ``` ## Contact & Support For questions or issues regarding this model: - Repository: [Potato-leaf-disease-prediction](https://github.com/krishna/Potato-leaf-disease-prediction) - Dataset Issues: Reference the Mendeley dataset documentation ## Disclaimer This model should be used as a reference tool and not as the sole basis for agricultural decisions. Always consult with agricultural experts for accurate disease diagnosis and treatment recommendations. --- **Model Version**: 1.0 **Last Updated**: May 2026 **Framework**: TensorFlow 2.x / Keras 3 **Python Version**: 3.8+