Pythonで画像のパワースペクトル

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

# 画像読込→グレースケール
img = cv2.imread("test.jpg")
img_gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 2次元FFT
img_fft = np.fft.fft2(img_gray)

# 象限の入れ替え。
img_fft =  np.fft.fftshift(img_fft)

# パワースペクトル
img_power = 20*np.log(np.abs(img_fft))

# 描画
plt.imshow(img_gray, cmap = 'gray')
plt.show()
plt.imshow(img_power , cmap = 'gray')
plt.show()

img_fft =  np.fft.fftshift(img_fft)