metadata
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
