| | --- |
| | imports: |
| | - "$import glob" |
| | - "$import numpy" |
| | - "$import os" |
| | input_channels: 1 |
| | output_classes: 3 |
| | arch_ckpt_path: "$@bundle_root + '/models/search_code_18590.pt'" |
| | arch_ckpt: "$torch.load(@arch_ckpt_path, map_location=torch.device('cuda'))" |
| | bundle_root: "." |
| | image_key: "image" |
| | output_dir: "$@bundle_root + '/eval'" |
| | output_ext: ".nii.gz" |
| | output_dtype: "$numpy.float32" |
| | output_postfix: "trans" |
| | separate_folder: true |
| | load_pretrain: true |
| | dataset_dir: "/workspace/data/msd/Task07_Pancreas" |
| | data_list_file_path: "$@bundle_root + '/configs/dataset_0.json'" |
| | datalist: "$monai.data.load_decathlon_datalist(@data_list_file_path, data_list_key='testing', |
| | base_dir=@dataset_dir)" |
| | device: "$torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')" |
| | dints_space: |
| | _target_: monai.networks.nets.TopologyInstance |
| | channel_mul: 1 |
| | num_blocks: 12 |
| | num_depths: 4 |
| | use_downsample: true |
| | arch_code: |
| | - "$@arch_ckpt['arch_code_a']" |
| | - "$@arch_ckpt['arch_code_c']" |
| | device: "$torch.device('cuda')" |
| | network_def: |
| | _target_: monai.networks.nets.DiNTS |
| | dints_space: "@dints_space" |
| | in_channels: "@input_channels" |
| | num_classes: "@output_classes" |
| | use_downsample: true |
| | node_a: "$torch.from_numpy(@arch_ckpt['node_a'])" |
| | network: "$@network_def.to(@device)" |
| | preprocessing: |
| | _target_: Compose |
| | transforms: |
| | - _target_: LoadImaged |
| | keys: "@image_key" |
| | - _target_: EnsureChannelFirstd |
| | keys: "@image_key" |
| | - _target_: Orientationd |
| | keys: "@image_key" |
| | axcodes: RAS |
| | - _target_: Spacingd |
| | keys: "@image_key" |
| | pixdim: |
| | - 1 |
| | - 1 |
| | - 1 |
| | mode: bilinear |
| | - _target_: ScaleIntensityRanged |
| | keys: "@image_key" |
| | a_min: -87 |
| | a_max: 199 |
| | b_min: 0 |
| | b_max: 1 |
| | clip: true |
| | - _target_: EnsureTyped |
| | keys: "@image_key" |
| | dataset: |
| | _target_: Dataset |
| | data: "@datalist" |
| | transform: "@preprocessing" |
| | dataloader: |
| | _target_: DataLoader |
| | dataset: "@dataset" |
| | batch_size: 1 |
| | shuffle: false |
| | num_workers: 4 |
| | inferer: |
| | _target_: SlidingWindowInferer |
| | roi_size: |
| | - 96 |
| | - 96 |
| | - 96 |
| | sw_batch_size: 4 |
| | overlap: 0.625 |
| | postprocessing: |
| | _target_: Compose |
| | transforms: |
| | - _target_: Activationsd |
| | keys: pred |
| | softmax: true |
| | - _target_: Invertd |
| | keys: pred |
| | transform: "@preprocessing" |
| | orig_keys: "@image_key" |
| | meta_key_postfix: meta_dict |
| | nearest_interp: false |
| | to_tensor: true |
| | - _target_: AsDiscreted |
| | keys: pred |
| | argmax: true |
| | - _target_: SaveImaged |
| | keys: pred |
| | meta_keys: pred_meta_dict |
| | output_dir: "@output_dir" |
| | output_ext: "@output_ext" |
| | output_dtype: "@output_dtype" |
| | output_postfix: "@output_postfix" |
| | separate_folder: "@separate_folder" |
| | handlers: |
| | - _target_: StatsHandler |
| | iteration_log: false |
| | evaluator: |
| | _target_: SupervisedEvaluator |
| | device: "@device" |
| | val_data_loader: "@dataloader" |
| | network: "@network" |
| | inferer: "@inferer" |
| | postprocessing: "@postprocessing" |
| | val_handlers: "@handlers" |
| | amp: true |
| | checkpointloader: |
| | _target_: CheckpointLoader |
| | load_path: "$@bundle_root + '/models/model.pt'" |
| | load_dict: |
| | model: "@network" |
| | initialize: |
| | - "$setattr(torch.backends.cudnn, 'benchmark', True)" |
| | - "$@checkpointloader(@evaluator) if @load_pretrain else None" |
| | run: |
| | - "$@evaluator.run()" |
| |
|
