add checkpoint

inference.json

@@ -1,6 +1,9 @@
 {
-    "dataset_dir": "/workspace/data/medical/spleen",
+    "dataset_dir": "/workspace/data/Task09_Spleen",
+    "ckpt_path": "/workspace/data//spleen_segmentation/models/model.pt",
+    "log": "$__import__('logging').basicConfig(level=20)",
     "device": "$torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')",
+    "datalist": "$list(sorted(__import__('glob').glob(@dataset_dir + '/imagesTs/*.nii.gz')))",
     "network_def": {
         "_target_": "UNet",
         "spatial_dims": 3,
@@ -34,6 +37,17 @@
                 "_target_": "EnsureChannelFirstd",
                 "keys": "image"
             },
+            {
+                "_target_": "Orientationd",
+                "keys": "image",
+                "axcodes": "RAS"
+            },
+            {
+                "_target_": "Spacingd",
+                "keys": "image",
+                "pixdim": [1.5, 1.5, 2.0],
+                "mode": "bilinear"
+            },
             {
                 "_target_": "ScaleIntensityRanged",
                 "keys": "image",
@@ -49,17 +63,9 @@
             }
         ]
     },
-    "datalist": {
-        "_target_": "DatasetFunc",
-        "data": "$@dataset_dir + '/dataset.json'",
-        "func": "$monai.data.load_decathlon_datalist",
-        "is_segmentation": true,
-        "data_list_key": "test",
-        "base_dir": "@dataset_dir"
-    },
     "dataset": {
         "_target_": "Dataset",
-        "data": "@datalist",
+        "data": "$[{'image': i} for i in @datalist]",
         "transform": "@preprocessing"
     },
     "dataloader": {
@@ -87,6 +93,15 @@
                 "keys": "pred",
                 "softmax": true
             },
+            {
+                "_target_": "Invertd",
+                "keys": "pred",
+                "transform": "@preprocessing",
+                "orig_keys": "image",
+                "meta_key_postfix": "meta_dict",
+                "nearest_interp": false,
+                "to_tensor": true
+            },
             {
                 "_target_": "AsDiscreted",
                 "keys": "pred",
@@ -95,11 +110,24 @@
             {
                 "_target_": "SaveImaged",
                 "keys": "pred",
-                "meta_keys": "image_meta_dict",
+                "meta_keys": "pred_meta_dict",
                 "output_dir": "eval"
             }
         ]
     },
+    "handlers": [
+        {
+            "_target_": "CheckpointLoader",
+            "_disabled_": "$not __import__('os').path.exists(@ckpt_path)",
+            "load_path": "@ckpt_path",
+            "load_dict": {"model": "@network"}
+        },
+        {
+            "_target_": "StatsHandler",
+            "_disabled_": "@log",
+            "iteration_log": false
+        }
+    ],
     "evaluator": {
         "_target_": "SupervisedEvaluator",
         "device": "@device",
@@ -107,6 +135,7 @@
         "network": "@network",
         "inferer": "@inferer",
         "postprocessing": "@postprocessing",
+        "val_handlers": "@handlers",
         "amp": false
     }
 }