surface_recto_3dunet / config.json
grgngl's picture
Recto-surface 3D segmentation U-Net (ps256) — checkpoints + model card (arXiv:2606.29085)
8ce8a46
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
}