Update monai/transforms/intensity/array.py

Thanks eric for the suggestion Co-authored-by: Eric Kerfoot <[email protected]> Signed-off-by: advcu <[email protected]>
Project-MONAI · Jan 14, 2025 · 3c58135 · 3c58135
1 parent 3656b99
commit 3c58135
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/monai/transforms/intensity/array.py b/monai/transforms/intensity/array.py
@@ -907,7 +907,8 @@ def __call__(self, img: NdarrayOrTensor) -> NdarrayOrTensor:
             if self.divisor is not None and len(self.divisor) != len(img):
                 raise ValueError(f"img has {len(img)} channels, but divisor has {len(self.divisor)} components.")
 
-            img, *_ = convert_data_type(img, dtype=torch.float32)
+            if not img.dtype.is_floating_point:
+                img, *_ = convert_data_type(img, dtype=torch.float32)
 
             for i, d in enumerate(img):
                 img[i] = self._normalize(  # type: ignore