Попытка использовать pyfftw - «dtypes выходного массива и входного массива не соответствуют допустимой схеме fftw». - PullRequest
Новая
       36

Попытка использовать pyfftw - «dtypes выходного массива и входного массива не соответствуют допустимой схеме fftw».

0 голосов
/ 31 мая 2019

Я использую pyfftw для простого FFT. Я не уверен, как использовать pyfftw, потому что я получаю сообщение об ошибке: 

KeyError                                  Traceback (most recent call last)
pyfftw\pyfftw.pyx in pyfftw.pyfftw.FFTW.__cinit__()

KeyError: (dtype('float64'), dtype('float64'))

During handling of the above exception, another exception occurred:

ValueError                                Traceback (most recent call last)
<ipython-input-31-dc32579f60f3> in <module>()
      3 start = time.time()
      4 b4 = numpy.zeros_like(a)
----> 5 fft = pyfftw.FFTW( a, b4, direction='FFTW_FORWARD', flags=('FFTW_MEASURE', ), threads=nthread, planning_timelimit=None )
      6 fft()
      7 end4 = time.time() - start

pyfftw\pyfftw.pyx in pyfftw.pyfftw.FFTW.__cinit__()

ValueError: Invalid scheme: The output array and input array dtypes do not correspond to a valid fftw scheme.

Это код, который я использую: 

import numpy as np
from scipy.signal import hilbert, chirp

duration = 1.0
fs = 400.0
samples = int(fs*duration)
t = np.arange(samples) / fs

signal = chirp(t, 20.0, t[-1], 100.0)
signal *= (1.0 + 0.5 * np.sin(2.0*np.pi*3.0*t) )

import time
import numpy
import pyfftw
import multiprocessing
nthread = multiprocessing.cpu_count()

a = signal

pyfftw.forget_wisdom()
start = time.time()
b4 = numpy.zeros_like(a)
fft = pyfftw.FFTW( a, b4, direction='FFTW_FORWARD', flags=('FFTW_MEASURE', ), threads=nthread, planning_timelimit=None )
fft()
end4 = time.time() - start

print(end4)

Любая идея о том, почему это не работает, будет принята с благодарностью

1 Ответ

1 голос
/ 31 мая 2019

Возможно, вы захотите использовать complex128 в качестве выходного типа, поскольку ДПФ действительного сигнала в общем случае является комплексным (если только действительный сигнал не симметричен).

См. https://hgomersall.github.io/pyFFTW/pyfftw/pyfftw.html#scheme-table

...