File size: 1,052 Bytes
38168b3 920053d 38168b3 c836248 38168b3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
---
license: apache-2.0
language: en
library_name: segmentation-models-pytorch
tags:
- segmentation
- pytorch
- computer-vision
- leaf-segmentation
- interactive-segmentation
datasets:
- LeafNet75/Annotated_Benchmarks400
pipeline_tag: image-segmentation
---
# Tomato-leaf-Segmentation
An Interactive model designed for auto-segmentation for domain specific: Tomato Leaf segmentation.
## Model Description
This model is a U-Net with a lightweight MobileNetV2 backbone, designed for interactive segmentation. It takes a 4-channel input (RGB image + user scribble) and outputs a binary segmentation mask.
This model was trained on the `LeafNet75/Annotated_Benchmarks400` dataset.
## Training Performance
The model was trained for 60 epochs with early stopping (patience=10).
### Final Validation Metrics:
- **Loss:** 0.0261
- **Dice Score:** 0.9475
- **IoU (Jaccard):** 0.9475
- **Precision:** 0.9711
- **Recall:** 0.9750
### Training History
### Sample Predictions
 |