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: