python - 使用 scipy.fftpack 出现内存错误，如何避免？

Question

我需要从 .tiff 医学图像中计算 Python 中的傅立叶系数。该代码产生内存错误：

for filename in glob.iglob ('*.tif'):
        imgfourier = scipy.misc.imread (filename, flatten = True)
        image = numpy.array([imgfourier])#make an array 
         # Take the fourier transform of the image.
        F1 = fftpack.fft2(image)
         # Now shift so that low spatial frequencies are in the center.
        F2 = fftpack.fftshift(F1)
         # the 2D power spectrum is:
        psd2D = np.abs(F2)**2
        print psd2D

任何帮助将非常感激！谢谢

Answer 1

我发现了这个讨论，他们在其中发现了内存泄漏scipy.fftpack，建议改用numpy.fftpackage。此外，您可以节省内存，避免中间变量：

import numpy as np
import glob
import scipy.misc
for filename in glob.iglob('*.tif'):
    imgfourier = scipy.misc.imread (filename, flatten = True)
    print np.abs(np.fft.fftshift(np.fft.fft2(np.array([imgfourier]))))**2