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

2023年6月10日 196次阅读

我一直在做一些使用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.

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

最佳答案 plt.imshow()需要整数.

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

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

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

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

image = image.astype(np.uint8)

点赞