File size: 2,124 Bytes
2657d76 8678d02 2657d76 | 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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | ---
license: mit
tags:
- image-segmentation
- multilabel
- unet
- pytorch
- medical-imaging
library_name: transformers
pipeline_tag: image-segmentation
---
# LN_segmentation_sweep
A unet model for multilabel image segmentation trained with sliding window approach.
## Model Description
- **Architecture:** unet
- **Input Channels:** 3
- **Output Classes:** 4
- **Base Filters:** 128
- **Window Size:** 128
- **Downsample Factor:** 1.0
### Model-Specific Parameters
## Training Configuration
| Parameter | Value |
|-----------|-------|
| Batch Size | 8 |
| Learning Rate | 7.77451918775676e-06 |
| Weight Decay | 0.00164040349077736 |
| Epochs | 100 |
| Patience | 10 |
| Dataset | GleghornLab/Semi-Automated_LN_Segmentation_10_11_2025 |
## Performance Metrics
| Metric | Mean | Class 0 | Class 1 | Class 2 | Class 3 |
|--------|------|--------|--------|--------|--------|
| Dice | 0.7781 | 0.7197 | 0.7614 | 0.7271 | 0.9040 |
| IoU | 0.6524 | 0.5708 | 0.6199 | 0.5929 | 0.8261 |
| F1 | 0.7781 | 0.7197 | 0.7614 | 0.7271 | 0.9040 |
| MCC | 0.7832 | 0.7308 | 0.7730 | 0.7440 | 0.8850 |
| ROC AUC | 0.9945 | 0.9945 | 0.9975 | 0.9917 | 0.9943 |
| PR AUC | 0.9076 | 0.8357 | 0.9108 | 0.9095 | 0.9743 |
## Usage
```python
import numpy as np
from model import MODEL_REGISTRY, SegmentationConfig
# Load model
config = SegmentationConfig.from_pretrained("aholk/LN_segmentation_sweep")
model = MODEL_REGISTRY["unet"].from_pretrained("aholk/LN_segmentation_sweep")
model.eval()
# Run inference on a full image with sliding window
image = np.random.rand(2048, 2048, 3).astype(np.float32) # Your image here
probs = model.predict_full_image(
image,
dim=128,
batch_size=16,
device="cuda" # or "cpu"
)
# probs shape: (num_classes, H, W) with values in [0, 1]
# Threshold to get binary masks
masks = (probs > 0.5).astype(np.uint8)
```
## Citation
If you use this model, please cite:
```bibtex
@software{windowz_segmentation,
title={Multilabel Image Segmentation with Sliding Window U-Net},
author={Gleghorn Lab},
year={2025},
url={https://github.com/GleghornLab/ComputerVision2}
}
```
|