Deep Learning pipeline for motor-imagery classification.

Last update: Oct 31, 2022

Overview

BCI-ToolBox

1. Introduction

BCI-ToolBox is deep learning pipeline for motor-imagery classification.
This repo contains five models: ShallowConvNet, DeepConvNet, EEGNet, FBCNet, BCI2021.
(BCI2021 is not an official name.)

2. Installation

Environment

Python == 3.7.10
PyTorch == 1.9.0
mne == 0.23.0
braindecode == 0.5.1
CUDA == 11.0

Create conda environment

conda install pytorch=1.9.0 cudatoolkit=11.1 -c pytorch -c nvidia
conda install numpy pandas matplotlib pyyaml ipywidgets
pip install torchinfo braindecode moabb mne

3. Directory structure

.
├── README.md
├── base
│   ├── constructor.py
│   └── layers.py
├── configs
│   ├── BCI2021
│   │   └── default.yaml
│   ├── DeepConvNet
│   │   └── default.yaml
│   ├── EEGNet
│   │   └── default.yaml
│   ├── FBCNet
│   │   └── default.yaml
│   ├── ShallowConvNet
│   │   └── default.yaml
│   └── demo
│       ├── arch.yaml
│       ├── bci2021.yaml
│       ├── test.yaml
│       ├── train.yaml
│       └── training_params.yaml
├── data_loader
│   ├── data_generator.py
│   ├── datasets
│   │   ├── __init__.py
│   │   ├── bnci2014.py
│   │   ├── cho2017.py
│   │   ├── folder_dataset.py
│   │   ├── openbmi.py
│   │   └── tmp_dataset.py
│   └── transforms.py
├── main.py
├── models
│   ├── BCI2021
│   │   ├── BCI2021.py
│   │   └── __init__.py
│   ├── DeepConvNet
│   │   ├── DeepConvNet.py
│   │   └── __init__.py
│   ├── EEGNet
│   │   ├── EEGNet.py
│   │   └── __init__.py
│   ├── FBCNet
│   │   ├── FBCNet.py
│   │   └── __init__.py
│   ├── ShallowConvNet
│   │   ├── ShallowConvNet.py
│   │   └── __init__.py
│   ├── __init__.py
│   └── model_builder.py
├── trainers
│   ├── __init__.py
│   ├── cls_trainer.py
│   └── trainer_maker.py
└── utils
    ├── calculator.py
    ├── painter.py
    └── utils.py

4. Dataset

Use braindecode

5. Get Started

We use W&B for experimental management tool.
Example for W&B Dashboard
Example for W&B Report

Create wandb_key.yaml file

Create wandb_key.yaml file in configs directory.
```
# wandb_key.yaml
key: WANDB API keys
```
WANDB API keys can be obtained from your W&B account settings.

train

Use W&B

python main.py --config_file=configs/demo/train.yaml

Not use W&B

python main.py --config_file=configs/demo/train.yaml --no_wandb

USE GPU

python main.py --config_file=configs/demo/train.yaml --device=0  # Use GPU 0
python main.py --config_file=configs/demo/train.yaml --device=1  # Use GPU 1
python main.py --config_file=configs/demo/train.yaml --device=2  # Use GPU 2

GPU numbers depend on your server.

USE Sweep

# W&B
sweep_file: configs/demo/training_params.yaml
project: Demo
tags: [train]

Add this block to config file for finding training parameters.

# W&B
sweep_file: configs/demo/arch.yaml
sweep_type: arch
project: Demo
tags: [train]

Add this block to config file for finding model architecture.

test

python main.py --config_file=configs/demo/test.yaml

5. References

ShallowConvNet [Paper] [Code]
DeepConvNet [Paper] [Code]
EEGNet [Paper] [Code]
FBCNet [Paper] [Code]
BCI2021 [Paper] [Code]

Deep Learning pipeline for motor-imagery classification.

Related tags

Overview

BCI-ToolBox

1. Introduction

2. Installation

Environment

Create conda environment

3. Directory structure

4. Dataset

5. Get Started

Create wandb_key.yaml file

train

test

5. References

Owner

DongHee

Code and real data for the paper "Counterfactual Temporal Point Processes", available at arXiv.

PyTorch implementation of Super SloMo by Jiang et al.

DAFNe: A One-Stage Anchor-Free Deep Model for Oriented Object Detection

A curated list of awesome papers for Semantic Retrieval (TOIS Accepted: Semantic Models for the First-stage Retrieval: A Comprehensive Review).

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2020

Codebase for Attentive Neural Hawkes Process (A-NHP) and Attentive Neural Datalog Through Time (A-NDTT)

Luminous is a framework for testing the performance of Embodied AI (EAI) models in indoor tasks.

Deeper insights into graph convolutional networks for semi-supervised learning

Boundary-preserving Mask R-CNN (ECCV 2020)

AniGAN: Style-Guided Generative Adversarial Networks for Unsupervised Anime Face Generation

Fader Networks: Manipulating Images by Sliding Attributes - NIPS 2017

Hyperbolic Hierarchical Clustering.

Deep deconfounded recommender (Deep-Deconf) for paper "Deep causal reasoning for recommendations"

Implementation of "Generalizable Neural Performer: Learning Robust Radiance Fields for Human Novel View Synthesis"

Source code for the BMVC-2021 paper "SimReg: Regression as a Simple Yet Effective Tool for Self-supervised Knowledge Distillation".

DIT is a DTLS MitM proxy implemented in Python 3. It can intercept, manipulate and suppress datagrams between two DTLS endpoints and supports psk-based and certificate-based authentication schemes (RSA + ECC).

Bayesian optimisation library developped by Huawei Noah's Ark Library

[ICCV 2021 Oral] Mining Latent Classes for Few-shot Segmentation

Code for "Typilus: Neural Type Hints" PLDI 2020

PyTorch implementation of paper "StarEnhancer: Learning Real-Time and Style-Aware Image Enhancement" (ICCV 2021 Oral)