การแปลง Cosine อย่างรวดเร็วผ่าน FFT

ฉันต้องการใช้การแปลงโคไซน์อย่างรวดเร็ว ฉันอ่านวิกิพีเดียว่ามีรุ่น DCT ที่รวดเร็วซึ่งคำนวณได้คล้ายกับ FFT ฉันพยายามอ่านMakhoul * paper ที่อ้างถึงสำหรับการใช้งาน FTPACK และ FFTW ที่ใช้ในScipyแต่ฉันไม่สามารถแยกอัลกอริทึมจริงได้ นี่คือสิ่งที่ฉันมี:

รหัส FFT:

def fft(x):
    if x.size ==1:
        return x
    N = x.size
    x0 = my_fft(x[0:N:2])
    x1 = my_fft(x[0+1:N:2])
    k = numpy.arange(N/2)
    e = numpy.exp(-2j*numpy.pi*k/N)
    l = x0 + x1 * e
    r = x0 - x1 * e  
    return numpy.hstack([l,r])

รหัส DCT:

def dct(x):
    k = 0
    N = x.size
    xk = numpy.zeros(N)
    for k in range(N):     
        for n in range(N):
            xn = x[n]
            xk[k] += xn*numpy.cos(numpy.pi/N*(n+1/2.0)*k)
    return xk 

ทดลองใช้ FCT:

def my_fct(x):
    if x.size ==1:
        return x
    N = x.size
    x0 = my_fct(x[0:N:2]) # have to be set to zero?
    x1 = my_fct(x[0+1:N:2])
    k = numpy.arange(N/2)
    n = # ???
    c = numpy.cos(numpy.pi/N*(n+1/2.0)*k)
    l = x0 #???
    r = x0 #???
    return numpy.hstack([l,r])

* J Makhoul "การแปลงโคไซน์ที่รวดเร็วในหนึ่งและสองมิติ" IEEE Trans Acoust Speech Sig พร 28 (1), 27-34 (1980)

Nxkarange(N)$[0, 1, 2, ..., N-1]$

พิมพ์ 2 DCT โดยใช้ 4N FFT และไม่มีกะการทำงาน

สัญญาณ[a, b, c, d]จะกลายเป็น

[0, a, 0, b, 0, c, 0, d, 0, d, 0, c, 0, b, 0, a].

จากนั้นใช้ FFT เพื่อรับสเปกตรัม

[A, B, C, D, 0, -D, -C, -B, -A, -B, -C, -D, 0, D, C, B]

จากนั้นให้ทิ้งทุกอย่างยกเว้นสิ่งแรก[A, B, C, D]และทำเสร็จแล้ว:

u = zeros(4 * N)
u[1:2*N:2] = x
u[2*N+1::2] = x[::-1]

U = fft(u)[:N]
return U.real

พิมพ์ 2 DCT โดยใช้มิร์เรอร์ 2N FFT (Makhoul)

[a, b, c, d][a, b, c, d, d, c, b, a][A, B, C, D, 0, D*, C*, B*][A, B, C, D]$e^{-j \pi k \over 2 N}$

y = empty(2*N)
y[:N] = x
y[N:] = x[::-1]

Y = fft(y)[:N]

Y *= exp(-1j*pi*k/(2*N))
return Y.real

พิมพ์ 2 DCT โดยใช้เบาะ 2N FFT (Makhoul)

[a, b, c, d][a, b, c, d, 0, 0, 0, 0][A, B, C, D, E, D*, C*, B*][A, B, C, D]$2e^{-j \pi k \over 2 N}$

y = zeros(2*N)
y[:N] = x

Y = fft(y)[:N]

Y *= 2 * exp(-1j*pi*k/(2*N))
return Y.real

พิมพ์ 2 DCT โดยใช้ N FFT (Makhoul)

[a, b, c, d, e, f][a, c, e, f, d, b][A, B, C, D, C*, B*]$2 e^{-j \pi k \over 2N}$

v = empty_like(x)
v[:(N-1)//2+1] = x[::2]

if N % 2: # odd length
    v[(N-1)//2+1:] = x[-2::-2]
else: # even length
    v[(N-1)//2+1:] = x[::-2]

V = fft(v)

V *= 2 * exp(-1j*pi*k/(2*N))
return V.real

บนเครื่องของฉันสิ่งเหล่านี้ล้วนมีความเร็วเท่ากันเนื่องจากการสร้างexp(-1j*pi*k/(2*N))ใช้เวลานานกว่า FFT : D

In [99]: timeit dct2_4nfft(a)
10 loops, best of 3: 23.6 ms per loop

In [100]: timeit dct2_2nfft_1(a)
10 loops, best of 3: 20.1 ms per loop

In [101]: timeit dct2_2nfft_2(a)
10 loops, best of 3: 20.8 ms per loop

In [102]: timeit dct2_nfft(a)
100 loops, best of 3: 16.4 ms per loop

In [103]: timeit scipy.fftpack.dct(a, 2)
100 loops, best of 3: 3 ms per loop

$x(n)$

ปล่อย

$y(n) = \Bigl\lbrace \begin{array}{ll} x(n), & n=0,1,...,N-1\\ x(2N - 1 - n), & n=N,N+1,...,2N-1 \end{array}$

DCT จะได้รับจากนั้น

$C(k) = \mathrm{Re}\Bigl\lbrace e^{-j\pi \frac{k}{2N}} FFT\lbrace y(n)\rbrace\Bigr\rbrace$

$2N$$y(n)$$x(n)$$x(n)$

นี่คือรหัสใน MATLAB

function C = fdct(x)
    N = length(x);
    y = zeros(1,2*N);
    y(1:N) = x;
    y(N+1:2*N) = fliplr(x);
    Y = fft(y);
    k=0:N-1;
    C = real(exp(-j.* pi.*k./(2*N)).*Y(1:N));

แก้ไข:

หมายเหตุ: สูตร DCT ที่ใช้คือ:

$C(k) = 2\sum_{n=0}^{N-1} x(n) \cos\left(\frac{\pi k}{2N}(2n+1) \right)$

มีหลายวิธีในการปรับขนาดของผลรวมดังนั้นจึงอาจไม่ตรงกับการใช้งานอื่น ๆ ตัวอย่างเช่น MATLAB ใช้:

$C(k) = w(k)\sum_{n=0}^{N-1} x(n) \cos\left(\frac{\pi k}{2N}(2n+1) \right)$

$w(0) = \sqrt\frac{1}{N}$$w(1...N-1) = \sqrt{\frac{2}{N}}$

คุณสามารถบัญชีเรื่องนี้ได้โดยการปรับขนาดผลลัพธ์

สำหรับการคำนวณทางวิทยาศาสตร์ที่แท้จริงปริมาณการใช้หน่วยความจำก็มีความสำคัญเช่นกัน ดังนั้นประเด็น N FFT จึงน่าสนใจยิ่งกว่าสำหรับฉัน สิ่งนี้เป็นไปได้เนื่องจาก Hermitian Symmetry ของสัญญาณ Makhoul อ้างอิงได้รับที่นี่ และมีอัลกอริทึมสำหรับการคำนวณ DCT และ IDCT หรือ DCT10 และ DCT01
http://ieeexplore.ieee.org/abstract/document/1163351/