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

test: test the plots of dwt and idwt

Browse Source
This commit is contained in:
Skudalen 2021-06-24 15:02:21 +02:00
parent 2d4bb65c99
commit 04498a79e6
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
Signal_prep.py
View File

@ -152,8 +152,8 @@ def plot_df(df:DataFrame):
plt.show() plt.show()
# Plots ndarrays after transformations # Plots ndarrays after transformations
def plot_trans(N_trans, y_trans): def plot_arrays(N, y):
plt.plot(N_trans, np.abs(y_trans)) plt.plot(N, np.abs(y))
plt.show() plt.show()
@ -161,9 +161,11 @@ def plot_trans(N_trans, y_trans):
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)
N = df.size N = np.array(range(int(df.iloc[:,1].size + 1)))
#plot_df(df) plot_df(df)
x, cA, cD = wavelet_db4_denoising(df) x, cA, cD = wavelet_db4_denoising(df)
#plot_trans(x, cA) plot_arrays(x, cA)
cA_filtered, cD = soft_threshold_filter(cA, cD) cA_filt, cD_filt = soft_threshold_filter(cA, cD)
plot_trans(x, cA_filtered) plot_arrays(x, cA_filt)
y_new_values = inverse_wavelet(cA, cD)
plot_arrays(N, y_new_values)