傅立叶反褶皱与numpy

时间:2018-04-14 16:52:30

标签: python numpy fft deconvolution

我试图使用傅里叶反卷积从信号中删除探头功能,但是我无法通过测试信号获得正确的输出。

t = np.zeros(30)
t = np.append(t, np.arange(0, 20, 0.1))

sigma = 2
mu = 5.
g = 1/np.sqrt(2*np.pi*sigma**2) * np.exp(-(np.arange(mu-3*sigma,mu+3*sigma,0.1)-mu)**2/(2*sigma**2))

def pad_signals(s1, s2):
    size = t.size +g.size - 1
    size = int(2 ** np.ceil(np.log2(size)))
    s1 = np.pad(s1, ((size-s1.size)//2, int(np.ceil((size-s1.size)/2))), 'constant', constant_values=(0, 0))
    s2 = np.pad(s2, ((size-s2.size)//2, int(np.ceil((size-s2.size)/2))), 'constant', constant_values=(0, 0))
    return s1, s2

def decon_fourier_ratio(signal, removed_signal):
    signal, removed_signal = pad_signals(signal, removed_signal)
    recovered = np.fft.fftshift(np.fft.ifft(np.fft.fft(signal)/np.fft.fft(removed_signal)))
    return np.real(recovered)

gt = (np.convolve(t, g, mode='full') / g.sum())[:230]
tr = decon_fourier_ratio(gt, g)

fig, ax = plt.subplots(nrows=2, ncols=2, sharex=True)
ax[0,0].plot(np.arange(0,np.fft.irfft(np.fft.rfft(t)).size), np.fft.irfft(np.fft.rfft(t)), label='thickness')
ax[0,1].plot(np.arange(0,np.fft.irfft(np.fft.rfft(g)).size), np.fft.irfft(np.fft.rfft(g)), label='probe shape')
ax[1,0].plot(np.arange(0,gt.size),gt, label='recorded signal')
ax[1,1].plot(np.arange(0,tr.size),tr, label='deconvolved signal')
plt.show()

enter image description here

上面的脚本创建了一个演示样本(t)和一个高斯形状的探针(g)。然后,它将它们卷积成信号gt,这是探测时样本的样子。我用pad_signals()将信号填充到最近的2 ^ N,以提高效率并修复任何非周期性。然后我尝试用decon_fourier_ratio()删除高斯探针。从图像可以清楚地看出,我没有恢复初始厚度梯度。任何关于解卷积为什么不起作用的想法?

注意:我也尝试过SciPy的去卷积。但是,此功能仅适用于某些宽度的高斯。

非常感谢任何帮助,

埃里克

1 个答案:

答案 0 :(得分:0)

你没有做完全卷积的任何理由?如果我将gt的结构更改为:

g /= g.sum()  # so the deconvolved signal has the same amplitude
gt = np.convolve(t, g, mode='full')

然后我得到以下情节:

enter image description here

我不能完全告诉你为什么你看到这种行为,除了部分卷积可能会改变频率内容。或者,如果要获得相同的行为并使用same,则可以使用零填充输入信号。