Upload README.md with huggingface_hub
Browse files
README.md
ADDED
|
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
+
license: mit
|
| 5 |
+
tags:
|
| 6 |
+
- image-classification
|
| 7 |
+
- pytorch
|
| 8 |
+
- convnext
|
| 9 |
+
- plant-disease
|
| 10 |
+
- agriculture
|
| 11 |
+
- edge-cloud
|
| 12 |
+
pipeline_tag: image-classification
|
| 13 |
+
model-index:
|
| 14 |
+
- name: ConvNeXt-Large Cloud Plant Classifier
|
| 15 |
+
results:
|
| 16 |
+
- task:
|
| 17 |
+
type: image-classification
|
| 18 |
+
name: Image Classification
|
| 19 |
+
dataset:
|
| 20 |
+
name: Plant Disease Classification Merged Dataset (88 Classes)
|
| 21 |
+
type: plant-disease-classification-merged-dataset
|
| 22 |
+
metrics:
|
| 23 |
+
- type: accuracy
|
| 24 |
+
value: 96.42
|
| 25 |
+
name: Validation Accuracy
|
| 26 |
+
---
|
| 27 |
+
|
| 28 |
+
# ConvNeXt-Large Cloud Plant Classifier
|
| 29 |
+
|
| 30 |
+
This repository hosts the **ConvNeXt-Large** cloud classifier model trained as part of the *Confidence-Aware Adaptive Edge–Cloud Framework for Reliable Plant Disease Diagnosis in Low-Connectivity Agricultural Environments*.
|
| 31 |
+
|
| 32 |
+
The cloud model serves as the high-capacity, heavy diagnostician tier of the cooperative inference pipeline. It is triggered dynamically by the edge node when a captured leaf image exhibits high epistemic uncertainty (vacuity $u > au_{vac}$) or low conformal confidence ($p_{max} < au_{conf}$).
|
| 33 |
+
|
| 34 |
+
## Model Architecture
|
| 35 |
+
- **Backbone**: ConvNeXt-Large (originally pre-trained on ImageNet-1K)
|
| 36 |
+
- **Classifier Head**: Modified Linear Layer projecting to 88 pathological crop classes
|
| 37 |
+
- **Input Dimension**: $384 imes 384$ pixels (higher resolution to capture fine-grained leaf spots)
|
| 38 |
+
|
| 39 |
+
## Training Setup & Parameters
|
| 40 |
+
- **Dataset**: Merged Plant Disease Dataset (88 unique classes spanning laboratory and field-captured folders).
|
| 41 |
+
- **Optimization Strategy**: Parameter-Efficient Fine-Tuning (PEFT) with convolutional stages 1–3 frozen, training only the 4th stage and classifier head.
|
| 42 |
+
- **Optimizer**: AdamW (Learning Rate: 1e-4, Weight Decay: 1e-3)
|
| 43 |
+
- **Loss Function**: Cross-Entropy Loss
|
| 44 |
+
- **Train/Val Split**: 85% Train / 15% Validation (Stratified per-folder split)
|
| 45 |
+
|
| 46 |
+
## Operational Performance (Epoch 7 Results)
|
| 47 |
+
Below are the recorded metrics from the training session:
|
| 48 |
+
|
| 49 |
+
| Metric | Score / Value | Description |
|
| 50 |
+
| :--- | :--- | :--- |
|
| 51 |
+
| **Training Loss** | `0.1039` | Cross-Entropy loss at epoch 7 |
|
| 52 |
+
| **Training Accuracy** | `96.47%` | Accuracy over the training split |
|
| 53 |
+
| **Validation Loss** | `0.1028` | Cross-Entropy loss on the held-out split |
|
| 54 |
+
| **Validation Accuracy** | **`96.42%`** | Calibrated validation classification rate |
|
| 55 |
+
|
| 56 |
+
## Integration & Deployment
|
| 57 |
+
The model checkpoint is designed to be hosted in a FastAPI-based Docker container on Hugging Face Spaces (Port 7860), responding to offloading queries via HTTP REST or binary JSON over MQTT v5 brokers.
|