FB-tCNN for SSVEP Classification

Here are the codes of the tCNN and FB-tCNN in the paper "Filter Bank Convolutional Neural Network for Short Time-Window Steady-State Visual Evoked Potential Classification"(DOI:10.1109/TNSRE.2021.3132162).

The related version information

1. Python == 3.7.0
2. Keras-gpu == 2.3.1
3. tensorflow-gpu == 2.1.0
4. scipy == 1.5.2
5. numpy == 1.19.2

   Training FB-tCNN for the public dataset

6. Download the code.
7. Download the public dataset and its paper.
8. Create a model folder to save the model.
9. Change the data and model folder paths in train and test files to your data and model folder paths.

   Training FB-tCNN for your own dataset

10. You need to design a new filter bank according to your dataset (Fundamental range of the SSVEP-EEG data). The filter bank details can refer to our paper.
11. The number of the sub-filters in the codes may be changed according to your own dataset.
12. The frequency of the target in the codes should be changed according to your own dataset.
13. The program that reads the data needs to be modified according to your own dataset.