From 297b126d93552eee284efc0ef59410d8b0c69446 Mon Sep 17 00:00:00 2001
From: LexvanGastel <l.vangastel@studen.utwente.nl>
Date: Tue, 1 Apr 2025 17:36:12 +0200
Subject: [PATCH] vraag11
---
main.py | 16 +++++++++-------
1 file changed, 9 insertions(+), 7 deletions(-)
diff --git a/main.py b/main.py
index 7f8caf8..faf91d7 100644
--- a/main.py
+++ b/main.py
@@ -103,30 +103,32 @@ def plot_emg_data(file_path):
num_samples = int(6 * sampling_rate)
num_samples = min(num_samples, data.shape[1])
time_vector = np.linspace(0,6,num_samples)
+
+
channel_rectus_femoris = data[chosen_muscles['rectus_fomoris'], :num_samples]
channel_tibialis_anterior = data[chosen_muscles['tibialis_anterior'], :num_samples]
+ signal_processor = SignalProcessing(data, sampling_rate)
+ filtered_rf = signal_processor.bandpass_filter(channel_rectus_femoris, 25, 300)
+ filtered_ta = signal_processor.bandpass_filter(channel_tibialis_anterior, 25, 300)
- freq_rf, fft_rf = compute_fft(channel_rectus_femoris[:num_samples], sampling_rate)
- freq_ta, fft_ta = compute_fft(channel_tibialis_anterior[:num_samples], sampling_rate)
+ freq_rf, fft_rf = compute_fft(filtered_rf, sampling_rate)
+ freq_ta, fft_ta = compute_fft(filtered_ta, sampling_rate)
plt.figure(figsize=(13,8))
plt.subplot(2, 2, 1)
- plt.plot(time_vector, channel_rectus_femoris)
+ plt.plot(time_vector, filtered_rf)
plt.title('EMG Data - Rectus femoris')
plt.xlabel('Time [Seconds]')
plt.ylabel('Amplitude')
plt.subplot(2, 2, 2)
- plt.plot(time_vector, channel_tibialis_anterior)
+ plt.plot(time_vector, filtered_ta)
plt.title('EMG Data - Tibialis Anterior')
plt.xlabel('Time [seconds]')
plt.ylabel('Amplitude')
-
- plt.tight_layout()
-
plt.subplot(2, 2, 3)
plt.plot(freq_rf, fft_rf)
plt.title('Power Spectrum - Rectus Femoris')
--
GitLab