Matplotlib imshow反转2D IFFT阵列的颜色

Question

我一直在做一些使用2D Scipy FFT对图像进行去卷积的工作。 然而，Matplotlib没有明显的原因是反转生成的IFFT阵列的颜色方案，即使RGB值是正确的。

import numpy as np
from scipy import fftpack
import matplotlib.pyplot as plt
import matplotlib.image as mpimg

image = mpimg.imread("C:/Users/-----/Desktop/image.jpg")
freq = fftpack.fft2(image)
IFFT = fftpack.ifft2(freq)
IFFT = IFFT.astype('float32')

plt.figure(1)
plt.imshow(image)

plt.figure(2)
plt.imshow(IFFT)

plt.show()

根据numpy.array_equal，IFFT数组和图像数组相等，但第二个数字的颜色图总是反转的。 请参见附图。 数组完全相同，没有指定其他颜色图，但我不得不使用以下方法手动反转所有内容：

for i in range(0, freq.shape[1]):
    for j in range (0, freq.shape[0]):
        for k in range(0,3):
            freq[j,i,k] = 255-freq[j,i,k]

我想知道转换为float32（或更早的uint32）的astype可能会改变某些东西，但由于数组相同，我不知道。

有任何想法吗？ 我还想弄清楚如何反转整个cmap，如果这是一个更有效的替代手动从数组中的每个条目减去255。

Answer 1

plt.imshow（）需要整数。

但是，我注意到即使两个整数数组看起来相同，它们也可能有不同类型的整数。

在我的例子中，当查看“右”数组中的整数类型（即颜色正确显示的那个）时，我得到了： class'numpy.uint8'

然而，当在plt.imshow（）中查找颜色被反转的数组中的整数类型时，我得到了： class'numpy.int64'

我需要做的就是将数组值转换为uint8：

image = image.astype（np.uint8）

Answer 2

一点也不。 您正在处理numpy数组，因此您可以：

 freq = 255 - freq

所以你的代码是：

import numpy as np
from scipy import fftpack
import matplotlib.pyplot as plt
import matplotlib.image as mpimg

image = mpimg.imread("EGZ68.jpg")
freq = fftpack.fft2(image)
freq = 255 - freq  # HERE IS THE CHANGE
IFFT = fftpack.ifft2(freq)
IFFT = IFFT.astype('float32')

plt.figure(1)
plt.imshow(image)

plt.figure(2)
plt.imshow(IFFT)

plt.show()

虽然我不确定你为什么要首先得到一个反转的色图。