Merge pull request #860 from scap3yvt/859-bug-noise-and-blur-enhanced…

…-fail-for-certain-images Ensure a float tensor is passed for augmentations
mlcommons · Apr 26, 2024 · 50f369f · 50f369f
2 parents dd88b88 + 7f47e5b
commit 50f369f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/GANDLF/data/augmentation/blur_enhanced.py b/GANDLF/data/augmentation/blur_enhanced.py
@@ -52,5 +52,5 @@ def calculate_std_ranges(self, image: torch.Tensor) -> Tuple[float, float]:
         std_ranges = self.std_original
         if self.std_original is None:
             # calculate the default std range based on 1.5% of the input image std - https://github.com/mlcommons/GaNDLF/issues/518
-            std_ranges = (0, 0.015 * torch.std(image.data).item())
+            std_ranges = (0, 0.015 * torch.std(image.data.float()).item())
         return self.parse_params(std_ranges, None, "std", min_constraint=0)
diff --git a/GANDLF/data/augmentation/noise_enhanced.py b/GANDLF/data/augmentation/noise_enhanced.py
@@ -66,5 +66,5 @@ def calculate_std_ranges(self, image: torch.Tensor) -> Tuple[float, float]:
         std_ranges = self.std_original
         if self.std_original is None:
             # calculate the default std range based on 1.5% of the input image std - https://github.com/mlcommons/GaNDLF/issues/518
-            std_ranges = (0, 0.015 * torch.std(image.data).item())
+            std_ranges = (0, 0.015 * torch.std(image.data.float()).item())
         return self._parse_range(std_ranges, "std", min_constraint=0)