Common transformations

ander-db · Oct 8, 2024 · ce67881 · ce67881
1 parent 7232245
commit ce67881
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/utils/common_transformations.py b/src/utils/common_transformations.py
@@ -1,5 +1,5 @@
-from src.utils.normalizations import NormalizeTransform
 from src.utils.scales import LogScaleTransform
+from src.utils.normalizations import NormalizeTransform
 from src.utils.transformations import RandomFlipRotateTransform
 
 
@@ -20,6 +20,12 @@ def __call__(self, input_tensor, target_tensor):
         return input_tensor, target_tensor
 
 
-TRANSFORM_LOG_NORM_FLIP = Compose(
+###################
+# TRANSFORMATIONS #
+###################
+
+TRANSFORM_LOG_NORM_DA = Compose(
     [LogScaleTransform(), NormalizeTransform(), RandomFlipRotateTransform()]
 )
+
+TRANSFORM_LOG_NORM = Compose([LogScaleTransform(), NormalizeTransform()])