search

tczhangzhi / PyTorch-ESCGAN

The official implementation of the paper "Beyond Mimicking Under-Represented Emotions: Deep Data Augmentation with Emotional Subspace Constraints for EEG-Based Emotion Recognition".
Other
16 stars 0 forks source link

ValueError: The following electrodes have overlapping positions, which causes problems during visualization: CB1, CB2 #2

Open Hanshuzheng149 opened 3 weeks ago

Hanshuzheng149 commented 3 weeks ago

求帮忙解决，这个错误怎么解决

tczhangzhi commented 3 weeks ago

Greeting,

The default 10-20 montage provided by mne doesn't include CB1 and CB2, whose locations are set to 0.0. Then, mne raises an error of overlap. You can either edit the montage locally (https://github.com/mne-tools/mne-python/blob/main/mne/channels/data/montages/standard_1020.elc) or set another montage:

import mne
import torch

from torcheeg.utils import plot_feature_topomap
from torcheeg.datasets.constants import SEED_CHANNEL_LIST

fake_eeg = torch.randn(62, 4)
img = plot_feature_topomap(fake_eeg,
                       channel_list=SEED_CHANNEL_LIST,
                       # see here  montage=mne.channels.make_standard_montage('standard_1005'))

You can also disable this visualization function since it is not related to the paper.