Update vertebra scripts (#1159)

Signed-off-by: Andres <[email protected]>

Signed-off-by: Andres <[email protected]>
Co-authored-by: SACHIDANAND ALLE <[email protected]>

Author: diazandr3s

sample-apps/radiology/lib/configs/localization_spine.py

@@ -107,7 +107,6 @@ def trainer(self) -> Optional[TrainTask]:
             load_path=load_path,
             publish_path=self.path[1],
             description="Train spine localization Model",
-            dimension=3,
             labels=self.labels,
         )
         return task

sample-apps/radiology/lib/configs/localization_vertebra.py

@@ -107,7 +107,6 @@ def trainer(self) -> Optional[TrainTask]:
             load_path=load_path,
             publish_path=self.path[1],
             description="Train vertebra localization Model",
-            dimension=3,
             labels=self.labels,
         )
         return task

sample-apps/radiology/lib/configs/segmentation_vertebra.py

@@ -106,7 +106,6 @@ def trainer(self) -> Optional[TrainTask]:
             load_path=load_path,
             publish_path=self.path[1],
             description="Train vertebra segmentation Model",
-            dimension=3,
             labels=self.labels,
         )
         return task

sample-apps/radiology/lib/trainers/localization_vertebra.py

@@ -118,7 +118,7 @@ def val_pre_transforms(self, context: Context):
             ScaleIntensityd(keys="image", minv=-1.0, maxv=1.0),
             CropForegroundd(
                 keys=("image", "label"),
-                source_key="image",
+                source_key="label",
                 margin=10,
                 k_divisible=[self.roi_size[0], self.roi_size[1], self.roi_size[2]],
             ),