configs/evaluate.json

{
    "validate#postprocessing": {
        "_target_": "Compose",
        "transforms": [
            {
                "_target_": "AsDiscreted",
                "keys": [
                    "pred",
                    "label"
                ],
                "argmax": [
                    true,
                    false
                ],
                "to_onehot": 2
            },
            {
                "_target_": "Lambdad",
                "keys": [
                    "pred"
                ],
                "func": "$lambda x : x[1:]"
            },
            {
                "_target_": "SaveImaged",
                "keys": "pred",
                "meta_keys": "pred_meta_dict",
                "output_dir": "@output_dir",
                "output_ext": ".png",
                "scale": 255,
                "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_iou",
                "val_mean_dice"
            ],
            "metric_details": [
                "val_iou",
                "val_mean_dice"
            ],
            "batch_transform": "$monai.handlers.from_engine(['image_meta_dict'])",
            "summary_ops": "*"
        }
    ],
    "evaluating": [
        "$setattr(torch.backends.cudnn, 'benchmark', True)",
        "$@validate#evaluator.run()"
    ]
}