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

fix: minor bugs

Browse Source
This commit is contained in:
Skudalen 2021-06-24 15:58:03 +02:00
parent 2098432d21
commit bdd476eb5e
2 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
Test_functions.py
View File

@ -1,6 +1,6 @@
from Handle_emg_data import CSV_handler, get_min_max_timestamp from Handle_emg_data import *
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
import Signal_prep from Signal_prep import *
def test_df_extraction(emg_nr): def test_df_extraction(emg_nr):
handler = CSV_handler() handler = CSV_handler()
@ -12,7 +12,7 @@ def test_df_extraction(emg_nr):
return subject1_left_emg1, emg_nr return subject1_left_emg1, emg_nr
def test_load_func(): def test_load_func():
test_dict = Signal_prep.load_user_emg_data() test_dict = load_user_emg_data()
subject2_container = test_dict[2] subject2_container = test_dict[2]
print(subject2_container.data_dict['left'][1]) print(subject2_container.data_dict['left'][1])
@ -29,24 +29,23 @@ def test_fft_prep():
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)
x, y, d = Signal_prep.prep_df_for_trans(df)
print(x)
print(y)
def test_plot_wavelet_both_ways(): def test_plot_wavelet_both_ways():
handler = CSV_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 = handler.get_xory_from_df('x', df) N = get_xory_from_df('x', df)
#plot_df(df) plot_df(df)
#print(len(N)) #print(len(N))
#print(len(get_xory_from_df('y', df))) #print(len(get_xory_from_df('y', df)))
x, cA, cD = handler.wavelet_db4_denoising(df) x, cA, cD = wavelet_db4_denoising(df)
#plot_arrays(x, cA) plot_arrays(x, cA)
#print(len(cA)) #print(len(cA))
cA_filt, cD_filt = soft_threshold_filter(cA, cD) cA_filt, cD_filt = soft_threshold_filter(cA, cD)
#plot_arrays(x, cA_filt) plot_arrays(x, cA_filt)
#print(len(cA_filt)) #print(len(cA_filt))
y_new_values = handler.inverse_wavelet(df, cA, cD) y_new_values = inverse_wavelet(df, cA, cD)
#print(len(y_new_values)) #print(len(y_new_values))
handler.plot_arrays(N, y_new_values) plot_arrays(N, y_new_values)
test_plot_wavelet_both_ways()

BIN
__pycache__/Signal_prep.cpython-38.pyc
View File

Binary file not shown.