Add imports and checkpoint
Browse files- inference.json +11 -5
inference.json
CHANGED
|
@@ -1,9 +1,14 @@
|
|
| 1 |
{
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
"dataset_dir": "/workspace/data/Task09_Spleen",
|
| 3 |
-
"ckpt_path": "/workspace/data
|
| 4 |
-
"
|
| 5 |
"device": "$torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')",
|
| 6 |
-
"datalist": "$list(sorted(
|
| 7 |
"network_def": {
|
| 8 |
"_target_": "UNet",
|
| 9 |
"spatial_dims": 3,
|
|
@@ -118,18 +123,19 @@
|
|
| 118 |
"handlers": [
|
| 119 |
{
|
| 120 |
"_target_": "CheckpointLoader",
|
| 121 |
-
"
|
|
|
|
| 122 |
"load_path": "@ckpt_path",
|
| 123 |
"load_dict": {"model": "@network"}
|
| 124 |
},
|
| 125 |
{
|
| 126 |
"_target_": "StatsHandler",
|
| 127 |
-
"_disabled_": "@log",
|
| 128 |
"iteration_log": false
|
| 129 |
}
|
| 130 |
],
|
| 131 |
"evaluator": {
|
| 132 |
"_target_": "SupervisedEvaluator",
|
|
|
|
| 133 |
"device": "@device",
|
| 134 |
"val_data_loader": "@dataloader",
|
| 135 |
"network": "@network",
|
|
|
|
| 1 |
{
|
| 2 |
+
"imports": [
|
| 3 |
+
"$import glob",
|
| 4 |
+
"$import os"
|
| 5 |
+
],
|
| 6 |
+
"cudnn_opt": "$setattr(torch.backends.cudnn, 'benchmark', True)",
|
| 7 |
"dataset_dir": "/workspace/data/Task09_Spleen",
|
| 8 |
+
"ckpt_path": "/workspace/data/tutorials/modules/bundles/spleen_segmentation/models/model.pt",
|
| 9 |
+
"download_ckpt": "$monai.apps.utils.download_url('https://huggingface.co/MONAI/example_spleen_segmentation/resolve/main/model.pt', @ckpt_path)",
|
| 10 |
"device": "$torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')",
|
| 11 |
+
"datalist": "$list(sorted(glob.glob(@dataset_dir + '/imagesTs/*.nii.gz')))",
|
| 12 |
"network_def": {
|
| 13 |
"_target_": "UNet",
|
| 14 |
"spatial_dims": 3,
|
|
|
|
| 123 |
"handlers": [
|
| 124 |
{
|
| 125 |
"_target_": "CheckpointLoader",
|
| 126 |
+
"_requires_": "@download_ckpt",
|
| 127 |
+
"_disabled_": "$not os.path.exists(@ckpt_path)",
|
| 128 |
"load_path": "@ckpt_path",
|
| 129 |
"load_dict": {"model": "@network"}
|
| 130 |
},
|
| 131 |
{
|
| 132 |
"_target_": "StatsHandler",
|
|
|
|
| 133 |
"iteration_log": false
|
| 134 |
}
|
| 135 |
],
|
| 136 |
"evaluator": {
|
| 137 |
"_target_": "SupervisedEvaluator",
|
| 138 |
+
"_requires_": "@cudnn_opt",
|
| 139 |
"device": "@device",
|
| 140 |
"val_data_loader": "@dataloader",
|
| 141 |
"network": "@network",
|