inverse_filtering.py

gistfile1.txt

# To add a new cell, type '# %%'
# To add a new markdown cell, type '# %% [markdown]'
# %%
import numpy as np
import matplotlib.pyplot as plt
import scipy.signal


h = np.zeros(128)
h[8] = 1
h[9] = 1

plt.plot(h)

x = np.array([1, 2, 1, 0, 4])

y = np.convolve(h, x)


# %%
H = np.fft.rfft(h)

plt.plot(np.abs(H))


# %%
plt.plot(x)
plt.plot(y)


# %%
d = np.zeros(len(h))
d[64] = 1
D = np.fft.rfft(d)

plt.plot(d)


# %%
# wiener filter
C = D * H.conj() / (np.abs(H)**2 + 0.05)

c = np.fft.irfft(C)

plt.plot(c)


# %%
plt.plot(np.abs(C))


# %%
C


# %%
plt.plot(np.abs(C*H))


# %%