使用OpenCV的FFT给出了两个矩阵？

Question

我正在使用numpy FFT来计算输入矩阵的频谱。我的样本输入是一个2D矩阵

A=[[1,2,3,4],[6,7,8,9]]

我的python代码是

FFT= np.fft.fftshift(np.fft.fft2(A))

输出

现在，我正在尝试使用Opencv的dft2功能。我得到类似于python的输出。但是我得到了不同维度的实数和想象输出的矩阵

我希望在像python这样的一个维度上得到一个具有实部和虚部的矩阵。我的代码使用OpenCV

dft=cv2.dft(np.float64(A),flags =  cv2.DFT_COMPLEX_OUTPUT|cv2.DFT_REAL_OUTPUT)
dft_shift = np.fft.fftshift(dft)

如何在python中使用OpenCV获得类似的结果。任何帮助将非常感谢

Answer 1

试试这个：

A=[[1,2,3,4],[6,7,8,9]]
dft=cv2.dft(np.float64(A),flags =  cv2.DFT_COMPLEX_OUTPUT)
dft_csv=dft[...,0]+1j*dft[...,1]
dft_np = np.fft.fft2(A)
assert (dft_np==dft_csv).all()
print(dft_np)
print(dft_csv)

输出为：

[[ 40.+0.j  -4.+4.j  -4.+0.j  -4.-4.j]
 [-20.+0.j   0.+0.j   0.+0.j   0.+0.j]]
[[ 40.+0.j  -4.+4.j  -4.+0.j  -4.-4.j]
 [-20.+0.j   0.+0.j   0.+0.j   0.+0.j]]

我希望这个迟到的回答能帮助新访客