README: code-only repo; unified-512 resolution-fair protocol (done)
Browse files
README.md
CHANGED
|
@@ -5,22 +5,28 @@ tags: [medical-imaging, segmentation, benchmark]
|
|
| 5 |
|
| 6 |
# GenSeg-Baselines
|
| 7 |
|
| 8 |
-
|
| 9 |
-
Companion to the [GenSegDataset](https://huggingface.co/datasets/MaybeRichard/GenSegDataset).
|
| 10 |
|
| 11 |
-
**
|
| 12 |
-
|
|
|
|
|
|
|
| 13 |
|
| 14 |
**Datasets:** cvc_clinicdb, kvasir_seg, fives, busi, refuge2, acdc, idridd, pannuke, isic2018, kits19.
|
| 15 |
|
| 16 |
-
**Metrics:** Dice, IoU, HD95, ASSD, Sensitivity, Specificity, Precision
|
|
|
|
| 17 |
|
| 18 |
-
##
|
| 19 |
-
|
| 20 |
-
-
|
| 21 |
-
|
| 22 |
|
| 23 |
-
##
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 5 |
|
| 6 |
# GenSeg-Baselines
|
| 7 |
|
| 8 |
+
Reproducible **code** for a 2D medical-image segmentation benchmark: **8 methods × 10 datasets × 3 seeds/folds, 7 metrics**, evaluated under a **unified resolution-fair protocol**. Companion to the [GenSegDataset](https://huggingface.co/datasets/MaybeRichard/GenSegDataset).
|
|
|
|
| 9 |
|
| 10 |
+
This is a **code-only** repository — trained checkpoints and the generated result tables are not hosted here.
|
| 11 |
+
|
| 12 |
+
**Methods:** UNet, UNet++, DeepLabV3+ (ResNet-50/ImageNet), Attention-UNet (from scratch),
|
| 13 |
+
TransUNet (R50-ViT-B/16, input 256), Swin-UNet (Swin-Tiny, input 224), nnU-Net v2 (250 ep), U-Mamba (UMambaBot, 100 ep).
|
| 14 |
|
| 15 |
**Datasets:** cvc_clinicdb, kvasir_seg, fives, busi, refuge2, acdc, idridd, pannuke, isic2018, kits19.
|
| 16 |
|
| 17 |
+
**Metrics (computed per image, then aggregated):** Dice, IoU, HD95, ASSD, Sensitivity, Specificity, Precision —
|
| 18 |
+
plus per-class Dice for the multi-class datasets and paired-Wilcoxon significance on per-image Dice.
|
| 19 |
|
| 20 |
+
## Resolution-fair protocol
|
| 21 |
+
Convolutional nets are trained at 512; the fixed-input transformers (Swin-UNet 224, TransUNet 256) and
|
| 22 |
+
nnU-Net / U-Mamba run at their native size; **every prediction and ground truth is resized to a common
|
| 23 |
+
512×512 before scoring**, so boundary metrics (HD95/ASSD, in pixels) are directly comparable across methods.
|
| 24 |
|
| 25 |
+
## Layout (code only)
|
| 26 |
+
- `code/framework/` — training/evaluation framework: `train.py`, `test.py`, `eval_at_res.py`,
|
| 27 |
+
`nnunet_eval.py`; `metrics/` (the 7 metrics + boundary distances); `models/` (SMP wrappers,
|
| 28 |
+
Attention-UNet, Swin/TransUNet wrappers, model registry); `report/aggregate.py` builds the summary
|
| 29 |
+
tables (per-dataset Dice/HD95/IoU, per-class Dice, Sensitivity/Precision, significance).
|
| 30 |
+
- `code/sota/{Swin-Unet,TransUNet}/` — upstream network definitions imported by the Swin-UNet / TransUNet wrappers.
|
| 31 |
+
- `code/scripts/` — reproduction scripts (unified-512 training & evaluation, nnU-Net / U-Mamba pipelines).
|
| 32 |
+
- `code/envs/` — conda environments (`seggen.yml`, `nnunet.yml`, `umamba.yml`).
|