NNUNet 3D Medical Image Segmentation with Multi-Fold Training

{ "model": { "type": "NNUNet", "plan_path": "data/preprocessed/nnUNetPlansv2.1_plans_3D.pkl", "stage": 0, "cascade": true }, "train_dataset": { "type": "MSDMultiFoldDataset", "plans_name": "nnUNetPlansv2.1_plans_3D.pkl", "dataset_root": "/", "result_dir": "/", "raw_data_dir": "data/raw_data", "decathlon_dir": "data/decathlon", "cropped_data_dir": "data/cropped", "preprocessed_dir": "data/preprocessed", "plan2d": false, "plan3d": true, "num_batches_per_epoch": 250, "num_folds": 5, "current_fold": 2, "stage": 0, "unpack_data": true, "cascade": true, "mode": "train" }, "val_dataset": { "type": "MSDMultiFoldDataset", "plans_name": "nnUNetPlansv2.1_plans_3D.pkl", "dataset_root": "/", "result_dir": "/", "raw_data_dir": "data/raw_data", "decathlon_dir": "data/decathlon", "cropped_data_dir": "data/cropped", "preprocessed_dir": "data/preprocessed", "num_batches_per_epoch": 50, "num_folds": 5, "current_fold": 2, "stage": 0, "plan2d": false, "plan3d": true, "unpack_data": true, "cascade": true, "mode": "val" }, "optimizer": { "type": "sgd", "momentum": 0.99, "weight_decay": 0.00003, "use_nesterov": true }, "lr_scheduler": { "type": "PolynomialDecay", "learning_rate": 0.01, "end_lr": 0, "power": 0.9 }, "loss": { "types": [ { "type": "MultipleLoss", "plan_path": "data/preprocessed/nnUNetPlansv2.1_plans_3D.pkl", "stage": 0, "losses": [ { "type": "DC_and_CE_loss", "do_bg": false, "batch_dice": false } ], "coef": [1.0] } ], "coef": [1] } }