| # ROI — 2D lung foreground segmentation. | |
| # Reproduces the bundled checkpoint at checkpoints/roi/best.pth | |
| # (val Dice = 0.96 on the unified split). | |
| task: roi | |
| data: | |
| data_root: ${DATA_ROOT} # contains ct_2d/ and roi_sem_seg_2d/ | |
| split_json: data/splits/unified.json | |
| preprocessing: | |
| target_size: [256, 256] | |
| model: | |
| name: segresnet | |
| spatial_dims: 2 | |
| in_channels: 1 | |
| out_channels: 1 # sigmoid | |
| init_filters: 16 | |
| blocks_down: [1, 2, 2, 4, 4] | |
| blocks_up: [1, 1, 1, 1] | |
| dropout_prob: 0.1 | |
| training: | |
| epochs: 100 | |
| batch_size: 16 | |
| samples_per_epoch: 20000 # subsample ~1/24 of ~475k slices | |
| lr: 1.0e-3 | |
| weight_decay: 1.0e-5 | |
| val_interval: 1 | |
| device: cuda | |
| amp: true | |
| amp_dtype: bfloat16 | |
| seed: 42 | |
| scheduler: | |
| T_max: 100 | |
| eta_min: 1.0e-6 | |
| checkpoints: | |
| dir: checkpoints/roi | |
| save_every: 10 | |
| logging: | |
| dir: runs/roi | |