From 4c677635248a9c0a455f2fbbdf55398c89481443 Mon Sep 17 00:00:00 2001
From: KumoLiu <yunl@nvidia.com>
Date: Mon, 25 Sep 2023 19:11:53 +0800
Subject: [PATCH] fix monai#7036

Signed-off-by: KumoLiu <yunl@nvidia.com>
---
 .../configs/inference.json                          | 12 +++++-------
 .../configs/metadata.json                           |  5 +++--
 .../endoscopic_tool_segmentation/configs/train.json | 13 ++++++++-----
 3 files changed, 16 insertions(+), 14 deletions(-)

diff --git a/models/endoscopic_tool_segmentation/configs/inference.json b/models/endoscopic_tool_segmentation/configs/inference.json
index 93dc7927..0f140d04 100644
--- a/models/endoscopic_tool_segmentation/configs/inference.json
+++ b/models/endoscopic_tool_segmentation/configs/inference.json
@@ -24,15 +24,13 @@
         "transforms": [
             {
                 "_target_": "LoadImaged",
-                "keys": [
-                    "image"
-                ]
+                "keys": "image",
+                "image_only": false
             },
             {
-                "_target_": "AsChannelFirstd",
-                "keys": [
-                    "image"
-                ]
+                "_target_": "EnsureChannelFirstd",
+                "keys": "image",
+                "channel_dim": -1
             },
             {
                 "_target_": "Resized",
diff --git a/models/endoscopic_tool_segmentation/configs/metadata.json b/models/endoscopic_tool_segmentation/configs/metadata.json
index c4b5af76..b70f30f7 100644
--- a/models/endoscopic_tool_segmentation/configs/metadata.json
+++ b/models/endoscopic_tool_segmentation/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.5.4",
+    "version": "0.5.5",
     "changelog": {
+        "0.5.5": "update AddChanneld with EnsureChannelFirstd and set image_only to False",
         "0.5.4": "fix the wrong GPU index issue of multi-node",
         "0.5.3": "remove error dollar symbol in readme",
         "0.5.2": "remove the CheckpointLoader from the train.json",
@@ -25,7 +26,7 @@
         "0.1.0": "complete the first version model package",
         "0.0.1": "initialize the model package structure"
     },
-    "monai_version": "1.2.0",
+    "monai_version": "1.3.0rc1",
     "pytorch_version": "1.13.1",
     "numpy_version": "1.22.2",
     "optional_packages_version": {
diff --git a/models/endoscopic_tool_segmentation/configs/train.json b/models/endoscopic_tool_segmentation/configs/train.json
index 28a182d8..0bab96da 100644
--- a/models/endoscopic_tool_segmentation/configs/train.json
+++ b/models/endoscopic_tool_segmentation/configs/train.json
@@ -53,15 +53,18 @@
                 "keys": [
                     "image",
                     "label"
-                ]
+                ],
+                "image_only": false
             },
             {
-                "_target_": "AsChannelFirstd",
-                "keys": "image"
+                "_target_": "EnsureChannelFirstd",
+                "keys": "image",
+                "channel_dim": -1
             },
             {
-                "_target_": "AddChanneld",
-                "keys": "label"
+                "_target_": "EnsureChannelFirstd",
+                "keys": "label",
+                "channel_dim": "no_channel"
             },
             {
                 "_target_": "Resized",