update output format of inference

configs/inference.json

@@ -89,11 +89,20 @@
                 "keys": "pred",
                 "threshold": 0.5
             },
+            {
+                "_target_": "Lambdad",
+                "keys": "pred",
+                "func": "$lambda x: torch.where(x[[2]] > 0, 4, torch.where(x[[0]] > 0, 1, torch.where(x[[1]] > 0, 2, 0)))"
+            },
             {
                 "_target_": "SaveImaged",
                 "keys": "pred",
                 "meta_keys": "pred_meta_dict",
-                "output_dir": "@output_dir"
+                "output_dir": "@output_dir",
+                "output_postfix": "seg",
+                "output_dtype": "uint8",
+                "resample": false,
+                "squeeze_end_dims": true
             }
         ]
     },

configs/metadata.json

@@ -1,7 +1,8 @@
 {
     "schema": "https://github.com/Project-MONAI/MONAI-extra-test-data/releases/download/0.8.1/meta_schema_20220324.json",
-    "version": "0.3.3",
+    "version": "0.3.4",
     "changelog": {
+        "0.3.4": "update output format of inference",
         "0.3.3": "update to use monai 1.0.1",
         "0.3.2": "enhance readme on commands example",
         "0.3.1": "fix license Copyright error",