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

fix: make the emg merger func so it makes one df per session

Browse Source
This commit is contained in:
Skudalen 2021-07-01 09:21:57 +02:00
parent aa27fb0aa4
commit 3eb838da54
3 changed files with 42 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
Handle_emg_data.py
View File

@ -444,10 +444,9 @@ class CSV_handler:
return self.data_container_dict return self.data_container_dict
# Retrieves df via the data_dict in the handler object # Retrieves df via the data_dict in the handler object
def get_df_from_data_dict(self, subject_nr, which_arm, round, emg_nr): def get_df_from_data_dict(self, subject_nr, which_arm, session, emg_nr):
data_type = self.data_type container:Data_container = self.data_container_dict.get(subject_nr)
container = self.data_container_dict.get(subject_nr) df = container.dict_list[session - 1].get(which_arm)[emg_nr - 1]
df = container.dict_list[round - 1].get(which_arm)[emg_nr - 1]
return df return df
# Loads in data to a CSV_handler. Choose data_type: hard, hardPP, soft og softPP as str. # Loads in data to a CSV_handler. Choose data_type: hard, hardPP, soft og softPP as str.
@ -497,38 +496,39 @@ class DL_data_handler:
def get_emg_list(self, subject_nr, session_nr) -> list: def get_emg_list(self, subject_nr, session_nr) -> list:
list_of_emgs = [] list_of_emgs = []
for emg_nr in range(8): for emg_nr in range(8):
df, _ = self.csv_handler.get_data(self.csv_handler, subject_nr, 'left', session_nr) df, _ = self.csv_handler.get_data(subject_nr, 'left', session_nr, emg_nr+1)
list_of_emgs.append(df) list_of_emgs.append(df)
for emg_nr in range(8): for emg_nr in range(8):
df, _ = self.csv_handler.get_data(self.csv_handler, subject_nr, 'right', session_nr) df, _ = self.csv_handler.get_data(subject_nr, 'right', session_nr, emg_nr+1)
list_of_emgs.append(df) list_of_emgs.append(df)
return list_of_emgs return list_of_emgs
def make_subj_sample(list_of_emgs): def make_subj_sample(self, list_of_emgs):
starting_point:DataFrame = list_of_emgs[0].rename(columns={'emg1':'emg'}) starting_point:DataFrame = list_of_emgs[0].rename(columns={'emg1':'emg'})
print(starting_point) tot_session_df_list = []
for i in range(2, 9):
tot_session_df:DataFrame = None
left_nr_remaining = len(list_of_emgs) - 1
for i in range(left_nr_remaining):
i += 1
emg_str = get_emg_str(i) emg_str = get_emg_str(i)
list_of_emgs[i].rename(columns={emg_str: 'emg'}, inplace=True) df = list_of_emgs[i-1].rename(columns={emg_str: 'emg'})
starting_point.append(list_of_emgs[i]) tot_session_df_list.append(df)
for i in range(1, 9):
emg_str = get_emg_str(i)
df = list_of_emgs[7+i].rename(columns={emg_str: 'emg'})
tot_session_df_list.append(df)
tot_session_df = starting_point.append(tot_session_df_list, ignore_index=True, sort=True)
return tot_session_df return tot_session_df
def store_samples(self, split_nr) -> None: def store_samples(self, split_nr) -> None:
for subject_nr in range(5): for subject_nr in range(5):
subj_samples = [] subj_samples = []
#session_df_list = []
for session_nr in range(4): for session_nr in range(4):
list_of_emg = self.get_emg_list(self.csv_handler, subject_nr, session_nr) list_of_emg = self.get_emg_list(subject_nr+1, session_nr+1)
tot_session_df = self.make_subj_sample(list_of_emg) tot_session_df, samplerate = self.make_subj_sample(list_of_emg)
#session_df_list.append(tot_session_df)
samples = np.array_split(tot_session_df, split_nr) samples = np.array_split(tot_session_df, split_nr)
for array in samples: for array in samples:
subj_samples.append(DataFrame(array)) df = DataFrame(array)
samplerate = get_samplerate(df)
subj_samples.append([df, samplerate])
self.samples_per_subject[subject_nr] = subj_samples self.samples_per_subject[subject_nr] = subj_samples

38
Present_data.py
View File

@ -143,9 +143,9 @@ def compare_with_wavelet_filter(data_frame):
# Input: CSV_handler # Input: CSV_handler
# Output: None --> Plot # Output: None --> Plot
def mfcc_3_plots_1_1_2(csv_handler:CSV_handler): def mfcc_3_plots_1_1_2(csv_handler:CSV_handler):
df1, samplerate1 = get_data(csv_handler, 1, 'left', 1, 1) df1, samplerate1 = csv_handler.get_data( 1, 'left', 1, 1)
df2, samplerate2 = get_data(csv_handler, 1, 'left', 2, 1) df2, samplerate2 = csv_handler.get_data( 1, 'left', 2, 1)
df3, samplerate3 = get_data(csv_handler, 2, 'left', 1, 1) df3, samplerate3 = csv_handler.get_data( 2, 'left', 1, 1)
#print(df1.head, samplerate1) #print(df1.head, samplerate1)
#print(df2.head, samplerate2) #print(df2.head, samplerate2)
#print(df3.head, samplerate3) #print(df3.head, samplerate3)
@ -162,9 +162,9 @@ def mfcc_3_plots_1_1_2(csv_handler:CSV_handler):
# Input: CSV_handler # Input: CSV_handler
# Output: None --> Plot # Output: None --> Plot
def mfcc_3_plots_3_3_4(csv_handler:CSV_handler): def mfcc_3_plots_3_3_4(csv_handler:CSV_handler):
df1, samplerate1 = get_data(csv_handler, 3, 'left', 1, 1) df1, samplerate1 = csv_handler.get_data(3, 'left', 1, 1)
df2, samplerate2 = get_data(csv_handler, 3, 'left', 2, 1) df2, samplerate2 = csv_handler.get_data(3, 'left', 2, 1)
df3, samplerate3 = get_data(csv_handler, 4, 'left', 1, 1) df3, samplerate3 = csv_handler.get_data(4, 'left', 1, 1)
#print(df1.head, samplerate1) #print(df1.head, samplerate1)
#print(df2.head, samplerate2) #print(df2.head, samplerate2)
#print(df3.head, samplerate3) #print(df3.head, samplerate3)
@ -178,14 +178,14 @@ def mfcc_3_plots_3_3_4(csv_handler:CSV_handler):
plot_3_mfcc(mfcc_feat1, label_1, mfcc_feat2, label_2, mfcc_feat3, label_3) plot_3_mfcc(mfcc_feat1, label_1, mfcc_feat2, label_2, mfcc_feat3, label_3)
def mfcc_all_emg_plots(csv_handler:CSV_handler): def mfcc_all_emg_plots(csv_handler:CSV_handler):
df1, samplerate1 = get_data(csv_handler, 1, 'left', 1, 1) df1, samplerate1 = csv_handler.get_data( 1, 'left', 1, 1)
df2, samplerate2 = get_data(csv_handler, 1, 'left', 1, 2) df2, samplerate2 = csv_handler.get_data( 1, 'left', 1, 2)
df3, samplerate3 = get_data(csv_handler, 1, 'left', 1, 3) df3, samplerate3 = csv_handler.get_data( 1, 'left', 1, 3)
df4, samplerate4 = get_data(csv_handler, 1, 'left', 1, 4) df4, samplerate4 = csv_handler.get_data( 1, 'left', 1, 4)
df5, samplerate5 = get_data(csv_handler, 1, 'left', 1, 5) df5, samplerate5 = csv_handler.get_data( 1, 'left', 1, 5)
df6, samplerate6 = get_data(csv_handler, 1, 'left', 1, 6) df6, samplerate6 = csv_handler.get_data( 1, 'left', 1, 6)
df7, samplerate7 = get_data(csv_handler, 1, 'left', 1, 7) df7, samplerate7 = csv_handler.get_data( 1, 'left', 1, 7)
df8, samplerate8 = get_data(csv_handler, 1, 'left', 1, 8) df8, samplerate8 = csv_handler.get_data( 1, 'left', 1, 8)
N1, mfcc_feat1 = mfcc_custom(df1, samplerate1, mfcc_windowsize, mfcc_stepsize) N1, mfcc_feat1 = mfcc_custom(df1, samplerate1, mfcc_windowsize, mfcc_stepsize)
N2, mfcc_feat2 = mfcc_custom(df2, samplerate2, mfcc_windowsize, mfcc_stepsize) N2, mfcc_feat2 = mfcc_custom(df2, samplerate2, mfcc_windowsize, mfcc_stepsize)
N3, mfcc_feat3 = mfcc_custom(df3, samplerate3, mfcc_windowsize, mfcc_stepsize) N3, mfcc_feat3 = mfcc_custom(df3, samplerate3, mfcc_windowsize, mfcc_stepsize)
@ -212,10 +212,14 @@ def mfcc_all_emg_plots(csv_handler:CSV_handler):
def main(): def main():
csv_handler = CSV_handler() csv_handler = CSV_handler()
load_data(csv_handler, 'soft') csv_handler.load_data('soft')
#print(csv_handler.get_data(1, 'left', 1, 1))
dl_data_handler = DL_data_handler(csv_handler) dl_data_handler = DL_data_handler(csv_handler)
dl_data_handler.store_samples(10) #dl_data_handler.store_samples(10)
print(dl_data_handler.samples_per_subject) #print(dl_data_handler.samples_per_subject)
emg_list = dl_data_handler.get_emg_list(1, 1)
df = dl_data_handler.make_subj_sample(emg_list)
print(df.head)
main() main()

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

Binary file not shown.