python – Matplotlib imshow反转2D IFFT数组的颜色

概述我一直在做一些使用2D Scipy FFT对图像进行去卷积的工作.然而,Matplotlib没有明显的原因是反转生成的IFFT阵列的颜色方案,即使RGB值是正确的. import numpy as npfrom scipy import fftpackimport matplotlib.pyplot as pltimport matplotlib.image as mpimgimage 我一直在做一些使用2D Scipy FFT对图像进行去卷积的工作.然而,Matplotlib没有明显的原因是反转生成的IFFT阵列的颜色方案,即使RGB值是正确的.

import numpy as npfrom scipy import fftpackimport matplotlib.pyplot as pltimport matplotlib.image as mpimgimage = 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,k]

我想知道转换为float32(或更早的uint32)的astype可能会改变某些东西,但由于数组相同,我不知道.

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

解决方法 plt.imshow()需要整数.

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

在我的例子中,当看到“右”数组中的整数类型(即颜色正确显示的那个)时,我得到了：
class’numpy.uint8′

而当在plt.imshow()中查找颜色被反转的数组中的整数类型时,我得到了：
class’numpy.int64′

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

image = image.astype(np.uint8)

总结

以上是内存溢出为你收集整理的python – Matplotlib imshow反转2D IFFT数组的颜色全部内容，希望文章能够帮你解决python – Matplotlib imshow反转2D IFFT数组的颜色所遇到的程序开发问题。

如果觉得内存溢出网站内容还不错，欢迎将内存溢出网站推荐给程序员好友。