MaybeRichard commited on
Commit
55acc31
·
verified ·
1 Parent(s): 1a18f22

README: code-only repo; unified-512 resolution-fair protocol (done)

Browse files
Files changed (1) hide show
  1. README.md +19 -13
README.md CHANGED
@@ -5,22 +5,28 @@ tags: [medical-imaging, segmentation, benchmark]
5
 
6
  # GenSeg-Baselines
7
 
8
- Baseline benchmark for 2D medical image segmentation: **8 methods x 10 datasets x 3 seeds/folds, 7 metrics**.
9
- Companion to the [GenSegDataset](https://huggingface.co/datasets/MaybeRichard/GenSegDataset).
10
 
11
- **Methods:** UNet, UNet++, DeepLabV3+ (ResNet-50/ImageNet), Attention-UNet (scratch),
12
- TransUNet (R50-ViT-B/16), Swin-UNet (Swin-Tiny), nnU-Net v2 (250ep), U-Mamba (UMambaBot, 100ep).
 
 
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 (+ efficiency).
 
17
 
18
- ## Layout
19
- - `code/` - baseline framework (train/test/aggregate), scripts, conda envs. *(Generative SegGen code excluded.)*
20
- - `results/` - per-run `metrics.json` + aggregated `summary.{html,csv,md,tex}` + `efficiency.md`.
21
- - `weights/` - curated checkpoints: best seed per (dataset, arch) for framework; best fold for nnU-Net / U-Mamba.
22
 
23
- ## Note
24
- These are the **256-px baseline** (confirmed). A resolution-fair re-evaluation (conv methods retrained at a
25
- higher per-dataset resolution; all methods scored at a common R so HD95 is comparable) is in progress and
26
- will be added later.
 
 
 
 
 
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`).