add ToCanonical

aramis-lab · Dec 27, 2024 · 1711484 · 1711484
1 parent 902d7dc
commit 1711484
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/clinicadl/transforms/config/enum.py b/clinicadl/transforms/config/enum.py
@@ -24,6 +24,7 @@ class ImplementedTransform(str, BaseEnum):
     Z_NORMALIZATION = "ZNormalization"
     MASK = "Mask"
     CROP_OR_PAD = "CropOrPad"
+    TO_CANONICAL = "ToCanonical"
     CLAMP = "Clamp"
     RESAMPLE = "Resample"
     RESIZE = "Resize"

diff --git a/clinicadl/transforms/config/factory.py b/clinicadl/transforms/config/factory.py
@@ -28,6 +28,7 @@
     PadConfig,
     ResampleConfig,
     ResizeConfig,
+    ToCanonicalConfig,
 )
 from .spatial_augmentations import (
     RandomAffineConfig,

diff --git a/clinicadl/transforms/config/spatial.py b/clinicadl/transforms/config/spatial.py
@@ -18,6 +18,7 @@
 
 __all__ = [
     "CropOrPadConfig",
+    "ToCanonicalConfig",
     "ResizeConfig",
     "ResampleConfig",
     "EnsureShapeMultipleConfig",
@@ -66,6 +67,16 @@ def check_shape(self):
         return self
 
 
+class ToCanonicalConfig(TransformConfig):
+    """Config class for ToCanonical transform."""
+
+    @computed_field
+    @property
+    def name(self) -> ImplementedTransform:
+        """The name of the transform."""
+        return ImplementedTransform.TO_CANONICAL
+
+
 class ResizeConfig(TransformConfig):
     """Config class for Resize transform."""