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",
|