| { |
| "imports": [ |
| "$import glob", |
| "$import os", |
| "$import ignite", |
| "$import json", |
| "$import urllib" |
| ], |
| "validate#handlers": [ |
| { |
| "_target_": "StatsHandler", |
| "iteration_log": false |
| }, |
| { |
| "_target_": "CheckpointLoader", |
| "load_path": "$@ckpt_dir + '/model.pt'", |
| "load_dict": { |
| "model": "@network" |
| } |
| }, |
| { |
| "_target_": "MetricsSaver", |
| "save_dir": "@output_dir", |
| "metrics": [ |
| "val_mean_dice", |
| "ar/dice", |
| "ve/dice", |
| "ur/dice", |
| "ki/dice", |
| "tu/dice", |
| "tu/haunsdorff", |
| "tu/surface" |
| ], |
| "metric_details": [ |
| "val_mean_dice" |
| ], |
| "batch_transform": "$monai.handlers.from_engine(['image_meta_dict'])", |
| "summary_ops": "*" |
| } |
| ], |
| "download_data": [ |
| "$import logging", |
| "$import zipfile", |
| "$import os", |
| "$logging.info('Downloading 298Mb data zip archive, please wait... (To see progress bar, download manually from https://disk.yandex.ru/d/pWEKt6D3qi3-aw , and extract data to bundle_root - ' + @bundle_root + ')')", |
| "$urllib.request.urlretrieve(json.loads(urllib.request.urlopen('https://cloud-api.yandex.net/v1/disk/public/resources/download?public_key=https%3A%2F%2Fdisk.yandex.ru%2Fd%2FpWEKt6D3qi3-aw').read())['href'], @bundle_root + '/AVUCTK_cases.zip')", |
| "$zipfile.ZipFile(os.path.join(@bundle_root, 'AVUCTK_cases.zip'), 'r').extractall(@bundle_root)", |
| "$os.remove(os.path.join(@bundle_root, 'AVUCTK_cases.zip'))", |
| "$logging.info('Data extracted to ' + @bundle_root)" |
| ], |
| "evaluate": [ |
| "$monai.utils.set_determinism(seed=42)", |
| "$setattr(torch.backends.cudnn, 'benchmark', True)", |
| "$@validate#evaluator.run()" |
| ] |
| } |
|
|
