/sleep_stage_classification_for_image

Primary LanguagePythonApache License 2.0Apache-2.0

Sleep Stage Classification

Standardized Image-Based Polysomnography Database and Deep Learning Algorithm for Sleep Stage Classification

1

Dataset

Our Dataset is currently not public.

Train

Convolutional Neural Network (ResNet 101)

python 1_train_conv.py

LSTM

python 2_train_lstm.py

Train with FOLD

Convolutional Neural Network (ResNet 101)

python 1_train_conv_fold.py

LSTM

python 2_train_lstm_fold.py

Test

python 3_test.py