| { | |
| "model_type": "monai_unet_3d", | |
| "architecture": { | |
| "spatial_dims": 3, | |
| "in_channels": 1, | |
| "out_channels": 2, | |
| "channels": [ | |
| 16, | |
| 32, | |
| 64, | |
| 128, | |
| 256, | |
| 512 | |
| ], | |
| "strides": [ | |
| 2, | |
| 2, | |
| 2, | |
| 2, | |
| 2, | |
| 2 | |
| ], | |
| "num_res_units": 2, | |
| "dropout": 0.2 | |
| }, | |
| "training": { | |
| "patch_size": [ | |
| 256, | |
| 256, | |
| 256 | |
| ], | |
| "loss_function": "DiceLoss", | |
| "optimizer": "Adam", | |
| "learning_rate": 0.0001, | |
| "weight_decay": 1e-05 | |
| }, | |
| "preprocessing": { | |
| "intensity_range": [ | |
| -1000, | |
| 4000 | |
| ], | |
| "output_range": [ | |
| 0, | |
| 1 | |
| ], | |
| "normalization": "ScaleIntensityRanged" | |
| } | |
| } |