EfficientNetV2-B0 plant disease model (54 classes, 98.0% test accuracy)

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+plantulus_doctor.keras filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,61 @@
+---
+license: mit
+library_name: keras
+pipeline_tag: image-classification
+tags:
+  - image-classification
+  - plant-disease
+  - agriculture
+  - tflite
+  - efficientnetv2
+---
+
+# Plantulus Doctor — plant disease classifier
+
+EfficientNetV2-B0 (ImageNet pre-trained) fine-tuned to recognise **54 plant/disease classes across 19 plants** from a photo of a single leaf. Live demo and JSON API: [KumarjitC/indoor-plant-disease-gradio-app](https://huggingface.co/spaces/KumarjitC/indoor-plant-disease-gradio-app).
+
+## Files
+
+| File | Use |
+|---|---|
+| `plantulus_doctor_fp16.tflite` | Float16-quantized TFLite model (~12 MB) for servers (LiteRT) and mobile |
+| `plantulus_doctor.keras` | Full Keras 3 model |
+| `classes.json` | Class labels in model output order |
+| `class_info.json` | Per class: label, plant, disease, `is_healthy`, display name |
+| `preprocessing.json` | Required input preprocessing |
+
+## Preprocessing
+
+1. Apply EXIF orientation and convert to RGB.
+2. Resize the shorter side to 256 px and centre-crop to 256×256.
+3. Centre-crop to 224×224.
+4. Feed float32 pixel values in **[0, 255]**. Do **not** divide by 255: the model normalizes internally.
+
+Input shape `[1, 224, 224, 3]`; output: softmax probabilities over the 54 classes in `classes.json` order.
+
+## Training data
+
+- [PlantVillage](https://www.kaggle.com/datasets/abdallahalidev/plantvillage-dataset) (colour images only)
+- [New Plant Diseases](https://www.kaggle.com/datasets/vipoooool/new-plant-diseases-dataset) (augmented PlantVillage)
+- [Indoor Plant Disease Detection](https://www.kaggle.com/datasets/abdulahad0296/indoor-plant-disease-detection-dataset)
+
+Images were deduplicated (MD5), and augmented copies were grouped by PlantVillage UUID and a rotation/flip-invariant perceptual hash so that copies of the same photo never appear in both train and test. Split ≈70/15/15, stratified by class.
+
+Training: two stages (frozen backbone head training, then full fine-tuning with frozen BatchNorm and cosine learning-rate decay), random-crop/flip/rotation/zoom/colour augmentation, square-root class weights, label smoothing 0.1, mixed precision. Because of label smoothing, confident predictions typically score around 0.9 rather than 1.0.
+
+## Evaluation (held-out test set, 16,697 images)
+
+| Subset | Images | Accuracy | Macro F1 |
+|---|---|---|---|
+| **All** | 16,697 | **98.04%** | **0.984** |
+| PlantVillage | 7,759 | 99.45% | 0.992 |
+| New Plant Diseases | 5,919 | 99.27% | 0.936 |
+| Indoor plants | 3,019 | 92.02% | 0.960 |
+
+Top-3 accuracy: 99.75%.
+
+## Limitations
+
+- Most training images are single leaves on plain backgrounds; expect lower accuracy on cluttered field or household photos (the indoor subset is the best indicator).
+- The model always predicts one of its 54 classes, even for unrelated images or unknown diseases; use the confidence score and top-3 output.
+- Not a substitute for expert diagnosis.

class_info.json

@@ -0,0 +1,380 @@
+[
+  {
+    "label": "aloe_anthracnose",
+    "plant": "Aloe",
+    "disease": "Anthracnose",
+    "is_healthy": false,
+    "display_name": "Aloe - Anthracnose"
+  },
+  {
+    "label": "aloe_healthy",
+    "plant": "Aloe",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Aloe - Healthy"
+  },
+  {
+    "label": "aloe_leafspot",
+    "plant": "Aloe",
+    "disease": "Leaf Spot",
+    "is_healthy": false,
+    "display_name": "Aloe - Leaf Spot"
+  },
+  {
+    "label": "aloe_rust",
+    "plant": "Aloe",
+    "disease": "Rust",
+    "is_healthy": false,
+    "display_name": "Aloe - Rust"
+  },
+  {
+    "label": "aloe_sunburn",
+    "plant": "Aloe",
+    "disease": "Sunburn",
+    "is_healthy": false,
+    "display_name": "Aloe - Sunburn"
+  },
+  {
+    "label": "apple_apple_scab",
+    "plant": "Apple",
+    "disease": "Scab",
+    "is_healthy": false,
+    "display_name": "Apple - Scab"
+  },
+  {
+    "label": "apple_black_rot",
+    "plant": "Apple",
+    "disease": "Black Rot",
+    "is_healthy": false,
+    "display_name": "Apple - Black Rot"
+  },
+  {
+    "label": "apple_cedar_apple_rust",
+    "plant": "Apple",
+    "disease": "Cedar Apple Rust",
+    "is_healthy": false,
+    "display_name": "Apple - Cedar Apple Rust"
+  },
+  {
+    "label": "apple_healthy",
+    "plant": "Apple",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Apple - Healthy"
+  },
+  {
+    "label": "blueberry_healthy",
+    "plant": "Blueberry",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Blueberry - Healthy"
+  },
+  {
+    "label": "cactus_dactylopius_opuntia",
+    "plant": "Cactus",
+    "disease": "Dactylopius Opuntia (Cochineal)",
+    "is_healthy": false,
+    "display_name": "Cactus - Dactylopius Opuntia (Cochineal)"
+  },
+  {
+    "label": "cactus_healthy",
+    "plant": "Cactus",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Cactus - Healthy"
+  },
+  {
+    "label": "cherry_including_sour_healthy",
+    "plant": "Cherry",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Cherry - Healthy"
+  },
+  {
+    "label": "cherry_including_sour_powdery_mildew",
+    "plant": "Cherry",
+    "disease": "Powdery Mildew",
+    "is_healthy": false,
+    "display_name": "Cherry - Powdery Mildew"
+  },
+  {
+    "label": "corn_maize_cercospora_leaf_spot_gray_leaf_spot",
+    "plant": "Corn (Maize)",
+    "disease": "Cercospora Leaf Spot (Gray Leaf Spot)",
+    "is_healthy": false,
+    "display_name": "Corn (Maize) - Cercospora Leaf Spot (Gray Leaf Spot)"
+  },
+  {
+    "label": "corn_maize_common_rust",
+    "plant": "Corn (Maize)",
+    "disease": "Common Rust",
+    "is_healthy": false,
+    "display_name": "Corn (Maize) - Common Rust"
+  },
+  {
+    "label": "corn_maize_healthy",
+    "plant": "Corn (Maize)",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Corn (Maize) - Healthy"
+  },
+  {
+    "label": "corn_maize_northern_leaf_blight",
+    "plant": "Corn (Maize)",
+    "disease": "Northern Leaf Blight",
+    "is_healthy": false,
+    "display_name": "Corn (Maize) - Northern Leaf Blight"
+  },
+  {
+    "label": "grape_black_rot",
+    "plant": "Grape",
+    "disease": "Black Rot",
+    "is_healthy": false,
+    "display_name": "Grape - Black Rot"
+  },
+  {
+    "label": "grape_esca_black_measles",
+    "plant": "Grape",
+    "disease": "Esca (Black Measles)",
+    "is_healthy": false,
+    "display_name": "Grape - Esca (Black Measles)"
+  },
+  {
+    "label": "grape_healthy",
+    "plant": "Grape",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Grape - Healthy"
+  },
+  {
+    "label": "grape_leaf_blight_isariopsis_leaf_spot",
+    "plant": "Grape",
+    "disease": "Leaf Blight (Isariopsis Leaf Spot)",
+    "is_healthy": false,
+    "display_name": "Grape - Leaf Blight (Isariopsis Leaf Spot)"
+  },
+  {
+    "label": "money_plant_bacterial_wilt_disease",
+    "plant": "Money Plant",
+    "disease": "Bacterial Wilt Disease",
+    "is_healthy": false,
+    "display_name": "Money Plant - Bacterial Wilt Disease"
+  },
+  {
+    "label": "money_plant_healthy",
+    "plant": "Money Plant",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Money Plant - Healthy"
+  },
+  {
+    "label": "money_plant_manganese_toxicity",
+    "plant": "Money Plant",
+    "disease": "Manganese Toxicity",
+    "is_healthy": false,
+    "display_name": "Money Plant - Manganese Toxicity"
+  },
+  {
+    "label": "orange_haunglongbing_citrus_greening",
+    "plant": "Orange",
+    "disease": "Huanglongbing (Citrus Greening)",
+    "is_healthy": false,
+    "display_name": "Orange - Huanglongbing (Citrus Greening)"
+  },
+  {
+    "label": "peach_bacterial_spot",
+    "plant": "Peach",
+    "disease": "Bacterial Spot",
+    "is_healthy": false,
+    "display_name": "Peach - Bacterial Spot"
+  },
+  {
+    "label": "peach_healthy",
+    "plant": "Peach",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Peach - Healthy"
+  },
+  {
+    "label": "pepper_bell_bacterial_spot",
+    "plant": "Bell Pepper",
+    "disease": "Bacterial Spot",
+    "is_healthy": false,
+    "display_name": "Bell Pepper - Bacterial Spot"
+  },
+  {
+    "label": "pepper_bell_healthy",
+    "plant": "Bell Pepper",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Bell Pepper - Healthy"
+  },
+  {
+    "label": "potato_early_blight",
+    "plant": "Potato",
+    "disease": "Early Blight",
+    "is_healthy": false,
+    "display_name": "Potato - Early Blight"
+  },
+  {
+    "label": "potato_healthy",
+    "plant": "Potato",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Potato - Healthy"
+  },
+  {
+    "label": "potato_late_blight",
+    "plant": "Potato",
+    "disease": "Late Blight",
+    "is_healthy": false,
+    "display_name": "Potato - Late Blight"
+  },
+  {
+    "label": "raspberry_healthy",
+    "plant": "Raspberry",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Raspberry - Healthy"
+  },
+  {
+    "label": "snake_plant_anthracnose",
+    "plant": "Snake Plant",
+    "disease": "Anthracnose",
+    "is_healthy": false,
+    "display_name": "Snake Plant - Anthracnose"
+  },
+  {
+    "label": "snake_plant_healthy",
+    "plant": "Snake Plant",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Snake Plant - Healthy"
+  },
+  {
+    "label": "snake_plant_leaf_withering",
+    "plant": "Snake Plant",
+    "disease": "Leaf Withering",
+    "is_healthy": false,
+    "display_name": "Snake Plant - Leaf Withering"
+  },
+  {
+    "label": "soybean_healthy",
+    "plant": "Soybean",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Soybean - Healthy"
+  },
+  {
+    "label": "spider_plant_fungal_leaf_spot",
+    "plant": "Spider Plant",
+    "disease": "Fungal Leaf Spot",
+    "is_healthy": false,
+    "display_name": "Spider Plant - Fungal Leaf Spot"
+  },
+  {
+    "label": "spider_plant_healthy",
+    "plant": "Spider Plant",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Spider Plant - Healthy"
+  },
+  {
+    "label": "spider_plant_leaf_tip_necrosis",
+    "plant": "Spider Plant",
+    "disease": "Leaf Tip Necrosis",
+    "is_healthy": false,
+    "display_name": "Spider Plant - Leaf Tip Necrosis"
+  },
+  {
+    "label": "squash_powdery_mildew",
+    "plant": "Squash",
+    "disease": "Powdery Mildew",
+    "is_healthy": false,
+    "display_name": "Squash - Powdery Mildew"
+  },
+  {
+    "label": "strawberry_healthy",
+    "plant": "Strawberry",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Strawberry - Healthy"
+  },
+  {
+    "label": "strawberry_leaf_scorch",
+    "plant": "Strawberry",
+    "disease": "Leaf Scorch",
+    "is_healthy": false,
+    "display_name": "Strawberry - Leaf Scorch"
+  },
+  {
+    "label": "tomato_bacterial_spot",
+    "plant": "Tomato",
+    "disease": "Bacterial Spot",
+    "is_healthy": false,
+    "display_name": "Tomato - Bacterial Spot"
+  },
+  {
+    "label": "tomato_early_blight",
+    "plant": "Tomato",
+    "disease": "Early Blight",
+    "is_healthy": false,
+    "display_name": "Tomato - Early Blight"
+  },
+  {
+    "label": "tomato_healthy",
+    "plant": "Tomato",
+    "disease": "Healthy",
+    "is_healthy": true,
+    "display_name": "Tomato - Healthy"
+  },
+  {
+    "label": "tomato_late_blight",
+    "plant": "Tomato",
+    "disease": "Late Blight",
+    "is_healthy": false,
+    "display_name": "Tomato - Late Blight"
+  },
+  {
+    "label": "tomato_leaf_mold",
+    "plant": "Tomato",
+    "disease": "Leaf Mold",
+    "is_healthy": false,
+    "display_name": "Tomato - Leaf Mold"
+  },
+  {
+    "label": "tomato_septoria_leaf_spot",
+    "plant": "Tomato",
+    "disease": "Septoria Leaf Spot",
+    "is_healthy": false,
+    "display_name": "Tomato - Septoria Leaf Spot"
+  },
+  {
+    "label": "tomato_spider_mites_two_spotted_spider_mite",
+    "plant": "Tomato",
+    "disease": "Spider Mites (Two-Spotted Spider Mite)",
+    "is_healthy": false,
+    "display_name": "Tomato - Spider Mites (Two-Spotted Spider Mite)"
+  },
+  {
+    "label": "tomato_target_spot",
+    "plant": "Tomato",
+    "disease": "Target Spot",
+    "is_healthy": false,
+    "display_name": "Tomato - Target Spot"
+  },
+  {
+    "label": "tomato_tomato_mosaic_virus",
+    "plant": "Tomato",
+    "disease": "Mosaic Virus",
+    "is_healthy": false,
+    "display_name": "Tomato - Mosaic Virus"
+  },
+  {
+    "label": "tomato_tomato_yellow_leaf_curl_virus",
+    "plant": "Tomato",
+    "disease": "Yellow Leaf Curl Virus",
+    "is_healthy": false,
+    "display_name": "Tomato - Yellow Leaf Curl Virus"
+  }
+]

classes.json

@@ -0,0 +1,56 @@
+[
+  "aloe_anthracnose",
+  "aloe_healthy",
+  "aloe_leafspot",
+  "aloe_rust",
+  "aloe_sunburn",
+  "apple_apple_scab",
+  "apple_black_rot",
+  "apple_cedar_apple_rust",
+  "apple_healthy",
+  "blueberry_healthy",
+  "cactus_dactylopius_opuntia",
+  "cactus_healthy",
+  "cherry_including_sour_healthy",
+  "cherry_including_sour_powdery_mildew",
+  "corn_maize_cercospora_leaf_spot_gray_leaf_spot",
+  "corn_maize_common_rust",
+  "corn_maize_healthy",
+  "corn_maize_northern_leaf_blight",
+  "grape_black_rot",
+  "grape_esca_black_measles",
+  "grape_healthy",
+  "grape_leaf_blight_isariopsis_leaf_spot",
+  "money_plant_bacterial_wilt_disease",
+  "money_plant_healthy",
+  "money_plant_manganese_toxicity",
+  "orange_haunglongbing_citrus_greening",
+  "peach_bacterial_spot",
+  "peach_healthy",
+  "pepper_bell_bacterial_spot",
+  "pepper_bell_healthy",
+  "potato_early_blight",
+  "potato_healthy",
+  "potato_late_blight",
+  "raspberry_healthy",
+  "snake_plant_anthracnose",
+  "snake_plant_healthy",
+  "snake_plant_leaf_withering",
+  "soybean_healthy",
+  "spider_plant_fungal_leaf_spot",
+  "spider_plant_healthy",
+  "spider_plant_leaf_tip_necrosis",
+  "squash_powdery_mildew",
+  "strawberry_healthy",
+  "strawberry_leaf_scorch",
+  "tomato_bacterial_spot",
+  "tomato_early_blight",
+  "tomato_healthy",
+  "tomato_late_blight",
+  "tomato_leaf_mold",
+  "tomato_septoria_leaf_spot",
+  "tomato_spider_mites_two_spotted_spider_mite",
+  "tomato_target_spot",
+  "tomato_tomato_mosaic_virus",
+  "tomato_tomato_yellow_leaf_curl_virus"
+]

plantulus_doctor.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4400fd8e889860c0e760c76e46cbb302687df6a460e8a38cf6247e5a576e615c
+size 24907594

plantulus_doctor_fp16.tflite

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3f9b1ef6e7bcc603c6844429ae4ac28607cbfb48fd7b1f856362c824a29a3e65
+size 11910620

preprocessing.json

@@ -0,0 +1,11 @@
+{
+  "input": "RGB image, float32 pixel values in [0, 255] (the model normalizes internally; do NOT divide by 255)",
+  "resize": "resize the shorter side to 256px, centre-crop to 256x256, then centre-crop to 224x224",
+  "input_shape": [
+    1,
+    224,
+    224,
+    3
+  ],
+  "output": "softmax probabilities over 54 classes, in the order of classes.json"
+}

test_metrics.json

@@ -0,0 +1,24 @@
+{
+  "test_images": 16697,
+  "accuracy": 0.9804156435287776,
+  "top3_accuracy": 0.9975444690662993,
+  "macro_f1": 0.9835331391496674,
+  "weighted_f1": 0.9804345892804046,
+  "per_source": {
+    "indoor_detection": {
+      "images": 3019,
+      "accuracy": 0.9201722424643922,
+      "macro_f1": 0.9603292403887314
+    },
+    "new_plant_diseases": {
+      "images": 5919,
+      "accuracy": 0.992735259334347,
+      "macro_f1": 0.9361751097090444
+    },
+    "plantvillage": {
+      "images": 7759,
+      "accuracy": 0.9944580487176182,
+      "macro_f1": 0.9924067738783012
+    }
+  }
+}