我使用numpy FFT从输入矩阵计算频谱。我的样本输入是一个2D矩阵
A=[[1,2,3,4],[6,7,8,9]]
我的python代码是
FFT= np.fft.fftshift(np.fft.fft2(A))
输出
现在,我正在尝试使用Opencv的dft2函数。我得到了类似于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获得类似的结果。任何帮助都将不胜感激
发布于 2021-05-27 03:26:55
试试这个:
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]]
我希望这篇最新的回答能帮助新的访问者。
https://stackoverflow.com/questions/37868859
复制相似问题