|
1 | 1 | _base_: |
2 | | - - base_banis_v2_erosion2.yaml |
| 2 | + - base_banis+.yaml |
3 | 3 |
|
4 | 4 | experiment_name: nisb_base_banis_plus_sdt |
5 | 5 | description: >- |
6 | | - banis+ (MedNeXt-L/k3) backbone combined with the v2 affinity+SDT two-head |
7 | | - setup and label erosion=2. Takes the higher-capacity affinity prediction of |
8 | | - base_banis+ and adds the auxiliary skeleton-aware distance transform head |
9 | | - from base_banis_v2 for false-merge reduction. |
| 6 | + banis+ (MedNeXt-L/k3) backbone with an affinity+SDT two-head and label |
| 7 | + erosion=2. The affinity head keeps the proven 6-channel r1+r10 supervision; |
| 8 | + the SDT head adds a skeleton-aware distance-transform signal for false-merge |
| 9 | + reduction. Decoding still uses only short-range affinity channels 0:3. |
| 10 | + Self-contained (the SDT two-head was previously provided by the now-removed |
| 11 | + base_banis_v2* configs). |
10 | 12 |
|
11 | 13 | save_path: outputs/nisb_base_banis_plus_sdt |
12 | 14 |
|
13 | 15 | default: |
14 | 16 | model: |
15 | | - arch: |
16 | | - profile: mednext_l |
17 | | - mednext: |
18 | | - size: L |
19 | | - kernel_size: 3 |
| 17 | + out_channels: 7 |
| 18 | + primary_head: aff |
| 19 | + heads: |
| 20 | + aff: |
| 21 | + out_channels: 6 |
| 22 | + num_blocks: 1 |
| 23 | + hidden_channels: 16 |
| 24 | + target_slice: "0:6" |
| 25 | + sdt: |
| 26 | + out_channels: 1 |
| 27 | + num_blocks: 1 |
| 28 | + hidden_channels: 16 |
| 29 | + target_slice: "6:7" |
| 30 | + loss: |
| 31 | + deep_supervision: false |
| 32 | + losses: |
| 33 | + - function: PerChannelBCEWithLogitsLoss |
| 34 | + weight: 1.0 |
| 35 | + pred_head: aff |
| 36 | + target_slice: "0:6" |
| 37 | + kwargs: |
| 38 | + auto_pos_weight: true |
| 39 | + max_pos_weight: 10.0 |
| 40 | + - function: SmoothL1Loss |
| 41 | + weight: 10.0 |
| 42 | + pred_head: sdt |
| 43 | + target_slice: "6:7" |
| 44 | + kwargs: |
| 45 | + tanh: true |
| 46 | + data: |
| 47 | + label_transform: |
| 48 | + targets: |
| 49 | + - name: affinity |
| 50 | + kwargs: |
| 51 | + offsets: ["1-0-0", "0-1-0", "0-0-1", "10-0-0", "0-10-0", "0-0-10"] |
| 52 | + affinity_mode: banis |
| 53 | + - name: skeleton_aware_edt |
| 54 | + kwargs: |
| 55 | + alpha: 0.8 |
| 56 | + bg_value: -1.0 |
| 57 | + inference: |
| 58 | + model: |
| 59 | + head: aff |
| 60 | + select_channel: [0, 1, 2] |
| 61 | + save_all_heads: true |
| 62 | + |
| 63 | +train: |
| 64 | + monitor: |
| 65 | + logging: |
| 66 | + scalar: |
| 67 | + loss: |
| 68 | + - train_loss_total_epoch |
| 69 | + - val_loss_total |
| 70 | + - train_loss_term_0_raw |
| 71 | + - train_loss_term_0_weighted |
| 72 | + - train_loss_term_1_raw |
| 73 | + - train_loss_term_1_weighted |
| 74 | + loss_every_n_steps: 100 |
| 75 | + images: |
| 76 | + head: all |
| 77 | + channel_mode: all |
0 commit comments