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

feat: add filtering func with SURE

Browse Source
This commit is contained in:
Skudalen 2021-06-24 13:55:39 +02:00
parent 3834141121
commit 360a0a28a3
1 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
Signal_prep.py
View File

@ -4,6 +4,7 @@ from pandas.core.frame import DataFrame
from scipy.fft import fft, fftfreq from scipy.fft import fft, fftfreq
import pywt import pywt
from scipy.signal import wavelets from scipy.signal import wavelets
from pyyawt import theselect
import Handle_emg_data as Handler import Handle_emg_data as Handler
@ -126,9 +127,12 @@ def denoise_signal_pywt(df:DataFrame):
cA, cD = pywt.dwt(y_values, wavelet) cA, cD = pywt.dwt(y_values, wavelet)
x = np.array(range(int(np.floor((y_values.size + wavelet.dec_len - 1) / 2)))) x = np.array(range(int(np.floor((y_values.size + wavelet.dec_len - 1) / 2))))
print(x) print(x)
return x, cA return x, cA, cD
#
def threshold_filter(cA, cD):
cA_filtered = theselect(cA, 'rigrsure')
return cA_filtered, cD
# Plots DataFrame objects # Plots DataFrame objects
def plot_df(df:DataFrame): def plot_df(df:DataFrame):
@ -144,8 +148,8 @@ def plot_trans(x_f, y_f):
handler = Handler.CSV_handler() handler = Handler.CSV_handler()
file = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1810/myoLeftEmg.csv" file = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1810/myoLeftEmg.csv"
df = handler.get_time_emg_table(file, 1) df = handler.get_time_emg_table(file, 1)
#plot_df(df) plot_df(df)
x_f, y_f = denoise_signal_pywt(df) x, cA, cD = denoise_signal_pywt(df)
#print(trans_df.info) plot_trans(x, cA)
plot_trans(x_f, y_f) cA_filtered, cD = np.ndarray(threshold_filter(cA, cD))
#''' plot_trans(x, cA_filtered)