[DLMED] update to adapt to MetaTensor and fixed a bug in customized component (#810)

Signed-off-by: Nic Ma <[email protected]>

Author: Nic-Ma

modules/bundle/custom_component/configs/custom_train.json

@@ -1,7 +1,7 @@
 {
-    "train#preprocessing#transforms#6":
+    "train#deterministic_transforms#0":
     {
-        "_target_": "scripts.custom_transforms.PrintEnsureTyped",
+        "_target_": "scripts.custom_transforms.LoadImagePrintd",
         "keys": ["image", "label"]
     }
 }

modules/bundle/custom_component/scripts/custom_transforms.py

@@ -10,20 +10,20 @@
 # limitations under the License.
 
 from monai.config import KeysCollection
-from monai.transforms import EnsureTyped
+from monai.transforms import LoadImaged
 
 
-class PrintEnsureTyped(EnsureTyped):
+class LoadImagePrintd(LoadImaged):
     """
-    Extend the `EnsureTyped` transform to print the image shape.
+    Transform to load image and print the image shape.
 
      Args:
         keys: keys of the corresponding items to be transformed.
 
     """
 
-    def __init__(self, keys: KeysCollection, data_type: str = "tensor") -> None:
-        super().__init__(keys, data_type=data_type)
+    def __init__(self, keys: KeysCollection) -> None:
+        super().__init__(keys)
 
     def __call__(self, data):
         d = dict(super().__call__(data=data))

modules/bundle/hybrid_programming/configs/data_loading.json

@@ -30,10 +30,6 @@
                 "b_min": 0,
                 "b_max": 1,
                 "clip": true
-            },
-            {
-                "_target_": "EnsureTyped",
-                "keys": "@image_key"
             }
         ]
     },

modules/bundle/spleen_segmentation/configs/train.json

@@ -63,10 +63,6 @@
                 "b_min": 0,
                 "b_max": 1,
                 "clip": true
-            },
-            {
-                "_target_": "EnsureTyped",
-                "keys": ["image", "label"]
             }
         ],
         "random_transforms": [