Update vertebra scripts (#1159)

Signed-off-by: Andres <[email protected]> Signed-off-by: Andres <[email protected]> Co-authored-by: SACHIDANAND ALLE <[email protected]>
Project-MONAI · Dec 3, 2022 · 4e72ec6 · 4e72ec6
1 parent 7183051
commit 4e72ec6
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 4 deletions.
diff --git a/sample-apps/radiology/lib/configs/localization_spine.py b/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
diff --git a/sample-apps/radiology/lib/configs/localization_vertebra.py b/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
diff --git a/sample-apps/radiology/lib/configs/segmentation_vertebra.py b/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
diff --git a/sample-apps/radiology/lib/trainers/localization_vertebra.py b/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]],
             ),