25 lines
816 B
Python
25 lines
816 B
Python
import mne
|
|
|
|
from utils.file_utils import get_epochs
|
|
|
|
|
|
def check_peaks():
|
|
"""
|
|
Sanity check for the "get_peaks" method
|
|
"""
|
|
import matplotlib.pyplot as plt
|
|
subject = '001'
|
|
folder = "../Dataset/n170/sub-" + subject + "/ses-n170/eeg/"
|
|
filepath = folder + "sub-" + subject + "_task-n170"
|
|
raw = mne.io.read_raw_fif(filepath + "_cleaned.fif")
|
|
epochs, _ = get_epochs(raw, [('face', 'intact')], picks='P7')
|
|
ch, latency, peak = epochs.average().get_peak(tmin=0.13, tmax=0.2, mode='neg', return_amplitude=True)
|
|
import numpy as np
|
|
plt.plot(epochs.times, np.squeeze(epochs.average().data.T))
|
|
plt.vlines([0.13, 0.2], -0.00001, 0.00001, colors='gray', linestyles='dotted')
|
|
plt.vlines(latency, -0.00001, 0.00001, colors='r', linestyles='dotted')
|
|
plt.show()
|
|
|
|
|
|
check_peaks()
|