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（）需要整数。

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

就我而言，在＆＃34;右边＆＃34;中查看整数类型时数组（即颜色显示正确的那个）我得到了： class＆＃39; numpy.uint8＆＃39;

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

我需要做的就是将数组值转换为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()

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