From 656307c3a2b592274c53cef25042441ee160b3d9 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fernando=20P=C3=A9rez-Garc=C3=ADa?= Date: Sun, 18 Jan 2026 22:28:28 +0000 Subject: [PATCH] Use abs() to compute maximum in k-space --- src/torchio/transforms/augmentation/intensity/random_spike.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/torchio/transforms/augmentation/intensity/random_spike.py b/src/torchio/transforms/augmentation/intensity/random_spike.py index ccf6c382..17e63457 100644 --- a/src/torchio/transforms/augmentation/intensity/random_spike.py +++ b/src/torchio/transforms/augmentation/intensity/random_spike.py @@ -153,8 +153,7 @@ def add_artifact( for index in indices: diff = index - mid_shape i, j, k = mid_shape + diff - # As of torch 1.7, "max is not yet implemented for complex tensors" - artifact = spectrum.cpu().numpy().max() * intensity_factor + artifact = spectrum.cpu().abs().max() * intensity_factor if self.invert_transform: spectrum[i, j, k] -= artifact else: