fix fftshift use in PS widget

brisvag · Feb 23, 2024 · 8d046c3 · 8d046c3
1 parent 00d41c9
commit 8d046c3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/blik/widgets/power_spectrum.py b/src/blik/widgets/power_spectrum.py
@@ -3,7 +3,7 @@
 import dask.array as da
 import numpy as np
 from magicgui import magic_factory
-from scipy.fft import fftn, fftshift
+from scipy.fft import fftn, fftshift, ifftshift
 
 if TYPE_CHECKING:
     import napari
@@ -24,7 +24,7 @@ def power_spectrum(
     axes = (1, 2) if image.metadata["stack"] else None
     raw = da.compute(image.data)[0]
     power_spectrum = np.abs(
-        fftshift(fftn(fftshift(raw, axes=axes), axes=axes), axes=axes)
+        fftshift(fftn(ifftshift(raw, axes=axes), axes=axes), axes=axes)
     )
     return (
         np.log(power_spectrum + 1),