diff --git a/main.py b/main.py index faf91d7b8ffc87b2177c3ad928f74a82bc71bcca..7afa61460bfab3f8543da3bfb72e4a35182eeca6 100644 --- a/main.py +++ b/main.py @@ -112,8 +112,11 @@ def plot_emg_data(file_path): 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(filtered_rf, sampling_rate) - freq_ta, fft_ta = compute_fft(filtered_ta, sampling_rate) + rectified_rf = np.abs(filtered_rf) + recitfied_ta = np.abs(filtered_ta) + + freq_rf, fft_rf = compute_fft(rectified_rf, sampling_rate) + freq_ta, fft_ta = compute_fft(recitfied_ta, sampling_rate) plt.figure(figsize=(13,8))