README.zh-CN.md

English | 简体中文

EEG-PyTorch

基于PyTorch的各种用于EEG分类任务的简单网络的实现。

已实现网络

• eeg-inception
• eeg-lstm
• eeg-gcn

使用方法

安装

使用 PYPI 安装。

pip install eeg-networks

如果不想安装包，也可以克隆本项目并安装依赖包。

git clone https://github.com/zrr1999/eeg-networks-pytorch
cd eeg-networks-pytorch
pip install -r requirements.txt

训练

可以通过以下命令训练模型

python train.py --model_name inception --device cpu --dataset_path ./GIB-UVA ERP-BCI.hdf5 --model_path ./weights

通过以下命令获取更多详细信息

python train.py --help

验证

可以通过以下命令验证模型

python val.py --model_path ./weights/last.pth --model_name inception

通过以下命令获取更多详细信息

python val.py --help

参考资料（Reference）

GIB-UVa ERP-BCI dataset