Fix README with Gradio SDK metadata
Browse files
README.md
CHANGED
|
@@ -1,3 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
# WoundNetB7 — Automated DFU Assessment Pipeline
|
| 2 |
|
| 3 |
End-to-end pipeline for Diabetic Foot Ulcer (DFU) analysis:
|
|
@@ -28,78 +40,14 @@ Input Image (DFU photograph)
|
|
| 28 |
-> 18% max group gap reduction (p < 10^-27)
|
| 29 |
```
|
| 30 |
|
| 31 |
-
## Quick Start
|
| 32 |
-
|
| 33 |
-
```python
|
| 34 |
-
from pipeline import WoundNetB7Pipeline
|
| 35 |
-
|
| 36 |
-
pipe = WoundNetB7Pipeline("models/")
|
| 37 |
-
result = pipe.analyze("path/to/dfu_image.png")
|
| 38 |
-
print(result.summary())
|
| 39 |
-
```
|
| 40 |
-
|
| 41 |
## Models
|
| 42 |
|
| 43 |
| Component | Architecture | Metric | Value |
|
| 44 |
|-----------|-------------|--------|-------|
|
| 45 |
| Segmentation | EfficientNet-B7 + UNet + ASPP + CBAM + TAM | Ulcer Dice | 0.927 |
|
| 46 |
-
| PWAT Item 3 | XGBoost + SMOTE | MAE / Adjacent | 0.87 / 74.9% |
|
| 47 |
| PWAT Item 7 | XGBoost + SMOTE | MAE / Adjacent | 0.61 / 89.3% |
|
| 48 |
| Fitzpatrick | ITA calibrated thresholds | Exact match | 86.9% |
|
| 49 |
|
| 50 |
-
## Fitzpatrick/ITA Calibrated Thresholds
|
| 51 |
-
|
| 52 |
-
| Type | ITA Range | Description |
|
| 53 |
-
|------|-----------|-------------|
|
| 54 |
-
| I | > 46.86 | Very Light |
|
| 55 |
-
| II | 34.25 - 46.86 | Light |
|
| 56 |
-
| III | 20.87 - 34.25 | Intermediate |
|
| 57 |
-
| IV | 3.57 - 20.87 | Tan |
|
| 58 |
-
| V | -28.38 - 3.57 | Brown |
|
| 59 |
-
| VI | < -28.38 | Dark |
|
| 60 |
-
|
| 61 |
-
## PWAT Debiasing Factors
|
| 62 |
-
|
| 63 |
-
Darker skin tones tend to overestimate tissue severity (especially Item 8: Periulcer Skin).
|
| 64 |
-
Correction factors reduce this bias:
|
| 65 |
-
|
| 66 |
-
| Type | P3 | P4 | P5 | P6 | P7 | P8 |
|
| 67 |
-
|------|-----|-----|-----|-----|-----|-----|
|
| 68 |
-
| I-II | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 |
|
| 69 |
-
| III | 0.0 | 0.0 | 0.0 | 0.0 | 0.0 | -0.1 |
|
| 70 |
-
| IV | -0.1 | -0.1 | 0.0 | 0.0 | 0.0 | -0.3 |
|
| 71 |
-
| V | -0.2 | -0.2 | -0.1 | 0.0 | 0.0 | -0.6 |
|
| 72 |
-
| VI | -0.3 | -0.3 | -0.2 | -0.1 | 0.0 | -0.9 |
|
| 73 |
-
|
| 74 |
-
## Dataset
|
| 75 |
-
|
| 76 |
-
- **Segmentation training**: 2,245 images (1,571 train / 336 val / 338 test)
|
| 77 |
-
- **Multiclass validation**: 461 images with 5-class expert masks
|
| 78 |
-
- **PWAT labels**: 1,321 images (920 expert GT + 401 AI consensus)
|
| 79 |
-
- **Fitzpatrick calibration**: 61 images with expert skin type annotations
|
| 80 |
-
|
| 81 |
-
## Deploy to Hugging Face
|
| 82 |
-
|
| 83 |
-
```bash
|
| 84 |
-
# Clone this repo to a HF Space
|
| 85 |
-
git clone https://huggingface.co/spaces/YOUR_USER/woundnetb7
|
| 86 |
-
# Copy models/ directory (or use git-lfs for large files)
|
| 87 |
-
git lfs install
|
| 88 |
-
git lfs track "*.pt" "*.pkl"
|
| 89 |
-
git add . && git commit -m "Initial deployment"
|
| 90 |
-
git push
|
| 91 |
-
```
|
| 92 |
-
|
| 93 |
## License
|
| 94 |
|
| 95 |
Research use only. Clinical deployment requires regulatory approval.
|
| 96 |
-
|
| 97 |
-
## Citation
|
| 98 |
-
|
| 99 |
-
```bibtex
|
| 100 |
-
@article{marquez2026woundnetb7,
|
| 101 |
-
title={WoundNetB7: Automated PWAT Protocol using Topological AI for Diabetic Foot Ulcers},
|
| 102 |
-
author={M{\'a}rquez-Sandoval, Marcelo},
|
| 103 |
-
year={2026}
|
| 104 |
-
}
|
| 105 |
-
```
|
|
|
|
| 1 |
+
---
|
| 2 |
+
title: WoundNetB7 DFU Analysis
|
| 3 |
+
emoji: 🩺
|
| 4 |
+
colorFrom: blue
|
| 5 |
+
colorTo: red
|
| 6 |
+
sdk: gradio
|
| 7 |
+
sdk_version: "5.29.0"
|
| 8 |
+
python_version: "3.11"
|
| 9 |
+
app_file: app.py
|
| 10 |
+
pinned: false
|
| 11 |
+
---
|
| 12 |
+
|
| 13 |
# WoundNetB7 — Automated DFU Assessment Pipeline
|
| 14 |
|
| 15 |
End-to-end pipeline for Diabetic Foot Ulcer (DFU) analysis:
|
|
|
|
| 40 |
-> 18% max group gap reduction (p < 10^-27)
|
| 41 |
```
|
| 42 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
## Models
|
| 44 |
|
| 45 |
| Component | Architecture | Metric | Value |
|
| 46 |
|-----------|-------------|--------|-------|
|
| 47 |
| Segmentation | EfficientNet-B7 + UNet + ASPP + CBAM + TAM | Ulcer Dice | 0.927 |
|
|
|
|
| 48 |
| PWAT Item 7 | XGBoost + SMOTE | MAE / Adjacent | 0.61 / 89.3% |
|
| 49 |
| Fitzpatrick | ITA calibrated thresholds | Exact match | 86.9% |
|
| 50 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
## License
|
| 52 |
|
| 53 |
Research use only. Clinical deployment requires regulatory approval.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|