灰度图像转换为频域图像

    img = nb.load(os.path.join(rootName,filename)).get_fdata()
    f = np.fft.fft2(img)
    fshift = np.fft.fftshift(f)
    # 这里构建振幅图的公式,一个y轴对称在正向递增的变换
    magnitude_spectrum = 20 * np.log(np.abs(fshift))