Fix evaluation

configs/evaluate.json

@@ -6,12 +6,27 @@
             {
                 "_target_": "Activationsd",
                 "keys": "pred",
-                "sigmoid": true
+                "softmax": true
             },
             {
                 "_target_": "AsDiscreted",
-                "keys": "pred",
-                "threshold": 0.5
+                "keys": [
+                    "pred",
+                    "label"
+                ],
+                "argmax": [
+                    true,
+                    false
+                ],
+                "to_onehot": 4
+            },
+            {
+                "_target_": "ToTensord",
+                "keys": [
+                    "pred",
+                    "label"
+                ],
+                "device": "@device"
             },
             {
                 "_target_": "SaveImaged",
@@ -39,11 +54,11 @@
             "_target_": "MetricsSaver",
             "save_dir": "@output_dir",
             "metrics": [
-                "val_mean_dice",
-                "val_acc"
+                "val_f1",
+                "val_accuracy"
             ],
             "metric_details": [
-                "val_mean_dice"
+                "val_f1"
             ],
             "batch_transform": "$monai.handlers.from_engine(['image_meta_dict'])",
             "summary_ops": "*"

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.0.3",
+    "version": "0.0.4",
     "changelog": {
+        "0.0.4": "Fix evaluation",
         "0.0.3": "Update to use MONAI 1.1.0",
         "0.0.2": "Update The Torch Vision Transform",
         "0.0.1": "initialize the model package structure"

configs/train.json

@@ -30,7 +30,7 @@
         "params": "[email protected]()",
         "lr": 0.0001
     },
-    "max_epochs": 5,
+    "max_epochs": 50,
     "train": {
         "preprocessing": {
             "_target_": "Compose",
@@ -259,7 +259,8 @@
                     "_target_": "SelectItemsd",
                     "keys": [
                         "image",
-                        "label"
+                        "label",
+                        "image_meta_dict"
                     ]
                 }
             ]