surface_recto_3dunet / config.json
grgngl's picture
Initial release — checkpoints accompanying Angelotti et al., Nature 2026
a5d0390 verified
Raw
History Blame Contribute Delete
1.67 kB
{
"squeeze_excitation": true,
"squeeze_excitation_type": "scse",
"patch_size": [
256,
256,
256
],
"train_patch_size": [
256,
256,
256
],
"batch_size": 2,
"train_batch_size": 2,
"targets": {
"surface": {
"valid_patch_value": 1,
"activation": "none",
"ignore_label": 2,
"losses": [
{
"name": "MedialSurfaceRecall",
"weight": 1.0
}
],
"out_channels": 2
}
},
"model_name": "ps256_bs2_msr_default",
"in_channels": 1,
"enable_deep_supervision": false,
"features_per_stage": [
32,
64,
128,
256,
320,
320,
320
],
"n_blocks_per_stage": [
1,
3,
4,
6,
6,
6,
6
],
"kernel_sizes": [
[
3,
3,
3
],
[
3,
3,
3
],
[
3,
3,
3
],
[
3,
3,
3
],
[
3,
3,
3
],
[
3,
3,
3
],
[
3,
3,
3
]
],
"strides": [
[
1,
1,
1
],
[
2,
2,
2
],
[
2,
2,
2
],
[
2,
2,
2
],
[
2,
2,
2
],
[
2,
2,
2
],
[
2,
2,
2
]
],
"pool_op_kernel_sizes": [
[
1,
1,
1
],
[
2,
2,
2
],
[
2,
2,
2
],
[
2,
2,
2
],
[
2,
2,
2
],
[
2,
2,
2
],
[
2,
2,
2
]
],
"separate_decoders": true
}