如何在opencv python中找到图像的相位谱

Question

我试图在 python 中应用 DFT 后找到图像的相位谱，这是我使用的代码。`

import numpy as np
import cv2
from matplotlib import pyplot as plt

img=cv2.imread('/content/drive/My Drive/IP assg2/im1.jpg')

img = cv2.cvtColor(sm1,cv2.COLOR_BGR2GRAY)

dft = cv2.dft(np.float32(img),flags = cv2.DFT_COMPLEX_OUTPUT)

`

我不知道如何从这里继续，因为我遇到的所有教程都与 MATLAB 相关。

Answer 1

你可以试试这个：

import numpy as np
import cv2
from matplotlib import pyplot as plt

img=cv2.imread('input.png')
img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
dft = np.fft.fft2(img)
dft_shift = np.fft.fftshift(dft)
phase_spectrum = np.angle(dft_shift)

ax1 = plt.subplot(1,2,1)
ax1.imshow(img, cmap='gray')

ax2 = plt.subplot(1,2,2)
ax2.imshow(phase_spectrum, cmap='gray')

plt.show()