├── .dockerignore ├── .gitignore ├── Dockerfile ├── LICENSE ├── README.md ├── config.py ├── helper_train.py ├── images ├── cluster.png └── semisupervised.png ├── models ├── __init__.py ├── model.py ├── resnet1d.py └── spectrogram_model.py ├── preprocessing ├── shhs │ ├── __init__.py │ ├── generate_shhs.py │ ├── preprocess_shhs.py │ ├── selected_shhs1_files.txt │ └── shhs_edfreader.py └── sleepedf │ ├── __init__.py │ ├── dhedfreader.py │ ├── generate_sleepedf.py │ └── preprocess_sleepedf.py ├── requirements.txt ├── train.py ├── utils ├── __init__.py ├── augmentations.py └── dataloader.py └── weights ├── shhs ├── cmc.pt ├── ours_diverse.pt ├── simple_fusion.pt ├── single_view.pt └── supervised_pretrain.pt └── sleepedf ├── cmc.pt ├── ours_diverse.pt ├── simple_fusion.pt ├── single_view.pt └── supervised_pretrain.pt /.dockerignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/.dockerignore -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/Dockerfile -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/config.py -------------------------------------------------------------------------------- /helper_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/helper_train.py -------------------------------------------------------------------------------- /images/cluster.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/images/cluster.png -------------------------------------------------------------------------------- /images/semisupervised.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/images/semisupervised.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/models/model.py -------------------------------------------------------------------------------- /models/resnet1d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/models/resnet1d.py -------------------------------------------------------------------------------- /models/spectrogram_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/models/spectrogram_model.py -------------------------------------------------------------------------------- /preprocessing/shhs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /preprocessing/shhs/generate_shhs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/preprocessing/shhs/generate_shhs.py -------------------------------------------------------------------------------- /preprocessing/shhs/preprocess_shhs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/preprocessing/shhs/preprocess_shhs.py -------------------------------------------------------------------------------- /preprocessing/shhs/selected_shhs1_files.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/preprocessing/shhs/selected_shhs1_files.txt -------------------------------------------------------------------------------- /preprocessing/shhs/shhs_edfreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/preprocessing/shhs/shhs_edfreader.py -------------------------------------------------------------------------------- /preprocessing/sleepedf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /preprocessing/sleepedf/dhedfreader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/preprocessing/sleepedf/dhedfreader.py -------------------------------------------------------------------------------- /preprocessing/sleepedf/generate_sleepedf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/preprocessing/sleepedf/generate_sleepedf.py -------------------------------------------------------------------------------- /preprocessing/sleepedf/preprocess_sleepedf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/preprocessing/sleepedf/preprocess_sleepedf.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/augmentations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/utils/augmentations.py -------------------------------------------------------------------------------- /utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/utils/dataloader.py -------------------------------------------------------------------------------- /weights/shhs/cmc.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/weights/shhs/cmc.pt -------------------------------------------------------------------------------- /weights/shhs/ours_diverse.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/weights/shhs/ours_diverse.pt -------------------------------------------------------------------------------- /weights/shhs/simple_fusion.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/weights/shhs/simple_fusion.pt -------------------------------------------------------------------------------- /weights/shhs/single_view.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/weights/shhs/single_view.pt -------------------------------------------------------------------------------- /weights/shhs/supervised_pretrain.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/weights/shhs/supervised_pretrain.pt -------------------------------------------------------------------------------- /weights/sleepedf/cmc.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/weights/sleepedf/cmc.pt -------------------------------------------------------------------------------- /weights/sleepedf/ours_diverse.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/weights/sleepedf/ours_diverse.pt -------------------------------------------------------------------------------- /weights/sleepedf/simple_fusion.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/weights/sleepedf/simple_fusion.pt -------------------------------------------------------------------------------- /weights/sleepedf/single_view.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/weights/sleepedf/single_view.pt -------------------------------------------------------------------------------- /weights/sleepedf/supervised_pretrain.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/likith012/mulEEG/HEAD/weights/sleepedf/supervised_pretrain.pt --------------------------------------------------------------------------------