Skip to content

Commit d114ef4

Browse files
Donglai Weiclaude
andcommitted
tutorials/neuron_nisb: drop base_banis_v2 configs; tweak base_banis+_sdt
Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
1 parent e698a1e commit d114ef4

3 files changed

Lines changed: 68 additions & 104 deletions

File tree

Lines changed: 68 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,77 @@
11
_base_:
2-
- base_banis_v2_erosion2.yaml
2+
- base_banis+.yaml
33

44
experiment_name: nisb_base_banis_plus_sdt
55
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).
1012
1113
save_path: outputs/nisb_base_banis_plus_sdt
1214

1315
default:
1416
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

tutorials/neuron_nisb/base_banis_v2.yaml

Lines changed: 0 additions & 77 deletions
This file was deleted.

tutorials/neuron_nisb/base_banis_v2_erosion2.yaml

Lines changed: 0 additions & 17 deletions
This file was deleted.

0 commit comments

Comments
 (0)