---
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
![Training Plots](training_plots.png)

### Sample Predictions
![Sample Predictions](sample_predictions.png)