πΏ Leaf Disease Classification with EfficientNetB0
Leaf-Disease-Classification-EfficientNetB0 is a deep learning-powered app that identifies 38 types of plant diseases from leaf images using a fine-tuned EfficientNetB0 model trained on the New Plant Diseases Dataset.
πΈ Upload an image or select an example below. Get instant predictions on plant disease type β with confidence score and class label.
π Demo
Deployed on Hugging Face Spaces:
π§ Model Details
- Model: EfficientNetB0 (Transfer Learning from ImageNet)
- Classes: 38 plant disease types (Apple, Corn, Potato, Tomato, etc.)
- Input Size: 224Γ224
- Preprocessing: EfficientNet
preprocess_input - Training:
- Optimizer: Adam
- Loss Function: Categorical Crossentropy
- Epochs: 20
- Validation Accuracy: ~97%
π Training Notebook (Kaggle)
The model was trained using TensorFlow and Keras on Kaggle.
π leaf-disease-classification-efficientnetb0-new1-1
Includes:
- Data Augmentation (Shear, Zoom, Flip)
- Early Stopping & Model Checkpointing
- Confusion Matrix Visualization
- Final
.kerasweights file exported
Features
- Classifies 38 different plant diseases
- Supports image upload and drag-and-drop
- Outputs top 3 predicted classes with confidence scores
- Lightweight & fast inference β powered by Gradio
π Folder Structure
Leaf-Disease-Project-CE180166/
βββ app.py # Gradio interface main script
βββ inference_utils.py # Helper functions (Load model & Predict)
βββ model/
β βββ final_model_20epochs.keras # Trained Model Weights
βββ assets/ # Demo/test images
β βββ Potato__Early_blight.jpg
β βββ ...
βββ requirements.txt # Dependencies
βββ README.md # Project Documentation
- Downloads last month
- 94