fix(transforms): fix norm trans bug

Author: okunator

cellseg_models_pytorch/transforms/albu_transforms/norm_transforms.py

@@ -45,6 +45,10 @@ def apply(self, image: np.ndarray, **kwargs) -> np.ndarray:
         """
         return minmax_normalize(image, self.amin, self.amax)
 
+    def get_transform_init_args_names(self):
+        """Get the names of the transformation arguments."""
+        return ("amin", "amax")
+
 
 class PercentileNormalization(ImageOnlyTransform):
     def __init__(
@@ -83,6 +87,10 @@ def apply(self, image: np.ndarray, **kwargs) -> np.ndarray:
         """
         return percentile_normalize(image, self.lower, self.upper)
 
+    def get_transform_init_args_names(self):
+        """Get the names of the transformation arguments."""
+        return ("lower", "upper")
+
 
 class ImgNormalization(ImageOnlyTransform):
     def __init__(
@@ -127,6 +135,10 @@ def apply(self, image: np.ndarray, **kwargs) -> np.ndarray:
         """
         return normalize(image, self.standardize, self.amin, self.amax)
 
+    def get_transform_init_args_names(self):
+        """Get the names of the transformation arguments."""
+        return ("amin", "amax", "standardize")
+
 
 def imgnorm_transform(**kwargs) -> List[ImageOnlyTransform]:
     """Return image-level normalization transform."""