ms762wu/EMG_Biometrics_2021
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

doc: add doc to inverse func

Browse Source
This commit is contained in:
Skudalen 2021-06-24 14:35:37 +02:00
parent c9b6230db8
commit 2d4bb65c99
1 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
Signal_prep.py
View File

@ -130,12 +130,21 @@ def wavelet_db4_denoising(df:DataFrame):
# Filters signal accordning to Stein's Unbiased Risk Estimate(SURE) # Filters signal accordning to Stein's Unbiased Risk Estimate(SURE)
def sure_threshold_filter(cA, cD): def sure_threshold_filter(cA, cD):
cA_filtered = pyyawt.theselect(cA, 'rigrsure') cA_filt = pyyawt.theselect(cA, 'rigrsure')
return cA_filtered, cD cD_filt = cD
return cA_filt, cD_filt
# soft filtering of wavelet trans with 0.25 lower percent
def soft_threshold_filter(cA, cD): def soft_threshold_filter(cA, cD):
cA_filtered = pywt.threshold(cA, 0.25 * cA.max()) cA_filt = pywt.threshold(cA, 0.25 * cA.max())
return cA_filtered, cD cD_filt = cD
return cA_filt, cD_filt
# Inverse dwt for brining denoise signal back to the time domain
def inverse_wavelet(cA_filt, cD_filt):
wavelet = pywt.Wavelet('db4')
y_new_values = pywt.idwt(cA_filt, cD_filt, wavelet)
return y_new_values
# Plots DataFrame objects # Plots DataFrame objects
def plot_df(df:DataFrame): def plot_df(df:DataFrame):
@ -147,10 +156,6 @@ def plot_trans(N_trans, y_trans):
plt.plot(N_trans, np.abs(y_trans)) plt.plot(N_trans, np.abs(y_trans))
plt.show() plt.show()
def inverse_wavelet(cA_filtered, cD):
wavelet = pywt.Wavelet('db4')
cA, cD = pywt.idwt(cA_filtered, cD, wavelet)
return cA, cD
#''' #'''
handler = Handler.CSV_handler() handler = Handler.CSV_handler()