Skip to content

Commit 34818ff

Browse files
author
Feng Wang
committed
change pyfftw to measure plane
1 parent ebd6fcb commit 34818ff

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

‎processing_library/fourier_transforms/fft_support.py‎

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -40,15 +40,15 @@ def fft(a):
4040
if (len(a.shape) == 4):
4141
b = pyfftw.interfaces.numpy_fft.fftshift(
4242
pyfftw.interfaces.numpy_fft.fft2(pyfftw.interfaces.numpy_fft.ifftshift(a, axes=[2, 3]),
43-
auto_align_input=False, planner_effort='FFTW_ESTIMATE'), axes=[2, 3])
43+
auto_align_input=False, planner_effort='FFTW_MEASURE'), axes=[2, 3])
4444
if (len(a.shape) == 5):
4545
b = pyfftw.interfaces.numpy_fft.fftshift(
4646
pyfftw.interfaces.numpy_fft.fft2(pyfftw.interfaces.numpy_fft.ifftshift(a, axes=[3, 4]),
47-
auto_align_input=False, planner_effort='FFTW_ESTIMATE'), axes=[3, 4])
47+
auto_align_input=False, planner_effort='FFTW_MEASURE'), axes=[3, 4])
4848
else:
4949
b = pyfftw.interfaces.numpy_fft.fftshift(
5050
pyfftw.interfaces.numpy_fft.fft2(pyfftw.interfaces.numpy_fft.ifftshift(a), auto_align_input=False,
51-
planner_effort='FFTW_ESTIMATE'))
51+
planner_effort='FFTW_MEASURE'))
5252
return b
5353

5454

@@ -73,19 +73,19 @@ def ifft(a):
7373
# a = pyfftw.byte_align(a)
7474
b = pyfftw.interfaces.numpy_fft.fftshift(
7575
pyfftw.interfaces.numpy_fft.ifft2(pyfftw.interfaces.numpy_fft.ifftshift(a, axes=[2, 3]),
76-
auto_align_input=False, planner_effort='FFTW_ESTIMATE'),
76+
auto_align_input=False, planner_effort='FFTW_MEASURE'),
7777
axes=[2, 3])
7878
if (len(a.shape) == 4):
7979
pass
8080
elif (len(a.shape) == 5):
8181
b = pyfftw.interfaces.numpy_fft.fftshift(
8282
pyfftw.interfaces.numpy_fft.ifft2(pyfftw.interfaces.numpy_fft.ifftshift(a, axes=[2, 3, 4]),
83-
auto_align_input=False, planner_effort='FFTW_ESTIMATE'),
83+
auto_align_input=False, planner_effort='FFTW_MEASURE'),
8484
axes=[2, 3, 4])
8585
else:
8686
b = pyfftw.interfaces.numpy_fft.fftshift(
8787
pyfftw.interfaces.numpy_fft.ifft2(pyfftw.interfaces.numpy_fft.ifftshift(a), auto_align_input=False,
88-
planner_effort='FFTW_ESTIMATE'))
88+
planner_effort='FFTW_MEASURE'))
8989
return b
9090

9191

0 commit comments

Comments
 (0)