configs/evaluate.json

{
    "validate#postprocessing": {
        "_target_": "Compose",
        "transforms": [
            {
                "_target_": "Activationsd",
                "keys": "pred",
                "sigmoid": true
            },
            {
                "_target_": "Invertd",
                "keys": "pred",
                "transform": "@validate#preprocessing",
                "orig_keys": "image",
                "meta_keys": "pred_meta_dict",
                "nearest_interp": false,
                "to_tensor": true,
                "device": "@validate#evaluator#device"
            },
            {
                "_target_": "AsDiscreted",
                "keys": "pred",
                "threshold": 0.5
            },
            {
                "_target_": "SplitDimd",
                "keys": [
                    "pred",
                    "label"
                ],
                "output_postfixes": [
                    "tc",
                    "wt",
                    "et"
                ]
            },
            {
                "_target_": "CopyItemsd",
                "keys": "pred",
                "names": "pred_combined",
                "times": 1
            },
            {
                "_target_": "Lambdad",
                "keys": "pred_combined",
                "func": "$lambda x: torch.where(x[[2]] > 0, 4, torch.where(x[[0]] > 0, 1, torch.where(x[[1]] > 0, 2, 0)))"
            },
            {
                "_target_": "SaveImaged",
                "keys": "pred_combined",
                "meta_keys": "pred_meta_dict",
                "output_dir": "@output_dir",
                "output_postfix": "seg",
                "output_dtype": "uint8",
                "resample": false,
                "squeeze_end_dims": true
            }
        ]
    },
    "validate#handlers": [
        {
            "_target_": "CheckpointLoader",
            "load_path": "$@ckpt_dir + '/model.pt'",
            "load_dict": {
                "model": "@network"
            }
        },
        {
            "_target_": "StatsHandler",
            "iteration_log": false
        },
        {
            "_target_": "MetricsSaver",
            "save_dir": "@output_dir",
            "metrics": [
                "val_mean_dice",
                "val_mean_dice_tc",
                "val_mean_dice_wt",
                "val_mean_dice_et"
            ],
            "metric_details": [
                "val_mean_dice"
            ],
            "batch_transform": "$monai.handlers.from_engine(['image_meta_dict'])",
            "summary_ops": "*"
        }
    ],
    "initialize": [
        "$setattr(torch.backends.cudnn, 'benchmark', True)"
    ],
    "run": [
        "$@validate#evaluator.run()"
    ]
}