From 3c58135fb7c545ab36d9f4df1d4c4fdfb4286d25 Mon Sep 17 00:00:00 2001 From: advcu <65158236+advcu987@users.noreply.github.com> Date: Tue, 14 Jan 2025 15:39:09 +0100 Subject: [PATCH] Update monai/transforms/intensity/array.py Thanks eric for the suggestion Co-authored-by: Eric Kerfoot <17726042+ericspod@users.noreply.github.com> Signed-off-by: advcu <65158236+advcu987@users.noreply.github.com> --- monai/transforms/intensity/array.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/monai/transforms/intensity/array.py b/monai/transforms/intensity/array.py index aefe5697b8..beb14b76e3 100644 --- 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