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

chore: move the plot test from Signal_prep

Browse Source 
to Test_functions
This commit is contained in:
Skudalen 2021-06-24 15:48:54 +02:00
parent dfc1b4173a
commit 2098432d21
2 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
Signal_prep.py
View File

@ -166,21 +166,3 @@ def plot_arrays(N, y):
plt.plot(N, np.abs(y)) plt.plot(N, np.abs(y))
plt.show() plt.show()
#'''
handler = Handler.CSV_handler()
file = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1810/myoLeftEmg.csv"
df = handler.get_time_emg_table(file, 1)
N = get_xory_from_df('x', df)
#plot_df(df)
print(len(N))
print(len(get_xory_from_df('y', df)))
x, cA, cD = wavelet_db4_denoising(df)
#plot_arrays(x, cA)
print(len(cA))
cA_filt, cD_filt = soft_threshold_filter(cA, cD)
#plot_arrays(x, cA_filt)
print(len(cA_filt))
y_new_values = inverse_wavelet(df, cA, cD)
print(len(y_new_values))
plot_arrays(N, y_new_values)

18
Test_functions.py
View File

@ -32,3 +32,21 @@ def test_fft_prep():
x, y, d = Signal_prep.prep_df_for_trans(df) x, y, d = Signal_prep.prep_df_for_trans(df)
print(x) print(x)
print(y) print(y)
def test_plot_wavelet_both_ways():
handler = CSV_handler()
file = "/Exp20201205_2myo_hardTypePP/HaluskaMarek_20201207_1810/myoLeftEmg.csv"
df = handler.get_time_emg_table(file, 1)
N = handler.get_xory_from_df('x', df)
#plot_df(df)
#print(len(N))
#print(len(get_xory_from_df('y', df)))
x, cA, cD = handler.wavelet_db4_denoising(df)
#plot_arrays(x, cA)
#print(len(cA))
cA_filt, cD_filt = soft_threshold_filter(cA, cD)
#plot_arrays(x, cA_filt)
#print(len(cA_filt))
y_new_values = handler.inverse_wavelet(df, cA, cD)
#print(len(y_new_values))
handler.plot_arrays(N, y_new_values)