AnguinusSculpturae / config.json
Bubenpo's picture
Add root config.json (pipeline summary; enables Hub download counting)
cd72cf8 verified
Raw
History Blame Contribute Delete
2.44 kB
{
"name": "mama-synth-foreground-stitched-lesion-folds-synthfolds-v1",
"description": "MAMA-SYNTH 2D pre-to-post contrast breast-MRI synthesis: fold-ensembled translation models composited along breast and tissue-support boundaries, followed by a multiplicative lesion-gain step.",
"framework": "nnunetv2",
"code_repository": "nnUNet-Mama-Synth (fork of MIC-DKFZ/nnUNet)",
"entry_point": "nnUNetv2_predict_mamasynth_translate_foreground_stitched_lesion_folds_synthfolds",
"configuration": "2d",
"plans_identifier": "mamaSynthPlans",
"network_class_name": "dynamic_network_architectures.architectures.unet.ResidualEncoderUNet",
"patch_size": [512, 512],
"spacing": [1.0, 1.0],
"normalization_schemes": ["ZScoreNormalization"],
"input_channels": {"0": "PRE"},
"num_networks": 14,
"parameters_per_network": 106000000,
"torch_min_version": "2.3.1",
"slots": {
"image": {
"role": "outside-breast translation",
"trainer": "nnUNetTrainerMamaSynthTranslationLPIPS_BS_48_MAEFinetune_ep_200",
"source_checkpoint": "checkpoint_best",
"folds": [0, 1, 2, 3],
"dataset": "Dataset625_Pre_Seg"
},
"lesion_image": {
"role": "inside-breast translation",
"trainer": "nnUNetTrainerMamaSynthTranslationLPIPSSSIMDiceTverskyA02B08_LPIPS05_BS_48_MAEFinetune",
"source_checkpoint": "checkpoint_best_dice",
"folds": [0, 1, 2, 3],
"dataset": "Dataset625_Pre_Seg"
},
"lesion_seg": {
"role": "lesion segmentation",
"trainer": "nnUNetTrainerMamaSynthLesionTverskyA02B08_BS_64_epoch_1000",
"source_checkpoint": "checkpoint_best",
"folds": [0, 1, 2, 3],
"dataset": "Dataset625_Pre_Seg"
},
"breast": {
"role": "breast segmentation",
"trainer": "nnUNetTrainerMamaSynthBreast_BS_32",
"source_checkpoint": "checkpoint_final",
"folds": [0],
"dataset": "Dataset625_Pre_Seg"
},
"foreground": {
"role": "tissue-support segmentation",
"trainer": "nnUNetTrainerMamaSynthForeground_BS_32",
"source_checkpoint": "checkpoint_final",
"folds": [0],
"dataset": "Dataset627_foreground"
}
},
"composite": {
"feather": 4,
"lesion_feather": 2,
"lesion_gain": 1.25,
"lesion_threshold": 0.5,
"lesion_step": 0.05,
"air_from": "pre",
"tta": true,
"mirror_axes": [0, 1],
"input_divisor": 64,
"expects_zscored_input": true
}
}