我一直在做一些使用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,k]
我想知道转换为float32(或更早的uint32)的astype可能会改变某些东西,但由于数组相同,我不知道.
有任何想法吗?我还想弄清楚如何反转整个cmap,如果这是一个更有效的替代手动从数组中的每个条目减去255.