This repository reproduces TinySleepNet for sleep stage prediction based on the signal channel EEG
using PyTorch and implements a new smaller and faster network called EmbedSleepNet.
# Environment Setup
1. Create a virtual environment with Python 3.8:
`virtualenv venv --python=python3.8`
2. Activate the environment:
`source venv/bin/activate` or `venv\Scripts\activate.bat` (for Windows)
3. Install dependencies:
`pip install -r requirements.txt`
4. Download and extract Sleep-EDF dataset from https://www.physionet.org/content/sleep-edfx/1.0.0/
5. Preprocess the data by running `python preprocess.py --data_path PATH_TO_sleep-cassete_FOLDER`
# Running
To start the training process simply run:
`python train.py --flavor=[embed, tiny]`, where tiny represents original TinySleepNet,
and embed represents newly introduced EmbedSleepNet.
You may additionally specify the number of epochs and model output name, for example:
`python train.py --flavor embed --epochs 450 --model-name model`
没有合适的资源?快使用搜索试试~ 我知道了~
温馨提示
【资源说明】 基于深度学习pytorch框架的单通道EEG睡眠分期项目Python源码.zip 基于深度学习pytorch框架的单通道EEG睡眠分期项目Python源码.zip 基于深度学习pytorch框架的单通道EEG睡眠分期项目Python源码.zip 基于深度学习pytorch框架的单通道EEG睡眠分期项目Python源码.zip基于深度学习pytorch框架的单通道EEG睡眠分期项目Python源码.zip 基于深度学习pytorch框架的单通道EEG睡眠分期项目Python源码.zip 基于深度学习pytorch框架的单通道EEG睡眠分期项目Python源码.zip 【备注】 1、该资源内项目代码都经过测试运行成功,功能ok的情况下才上传的,请放心下载使用! 2、本项目适合计算机相关专业(如计科、人工智能、通信工程、自动化、电子信息等)的在校学生、老师或者企业员工下载使用,也适合小白学习进阶,当然也可作为毕设项目、课程设计、作业、项目初期立项演示等。 3、如果基础还行,也可在此代码基础上进行修改,以实现其他功能,也可直接用于毕设、课设、作业等。 欢迎下载,沟通交流,互相学习,共同进步!
资源推荐
资源详情
资源评论
收起资源包目录
基于深度学习pytorch框架的单通道EEG睡眠分期项目Python源码.zip (21个子文件)
README.md 52B
EmbedSleepNet
preprocess.py 5KB
benchmark.py 368B
lightning_wrapper.py 3KB
LICENSE 1KB
dataset.py 3KB
model.py 3KB
.idea
misc.xml 199B
inspectionProfiles
Project_Default.xml 942B
profiles_settings.xml 174B
modules.xml 288B
EmbedSleepNet-main.iml 409B
.gitignore 182B
requirements.txt 179B
.gitignore 2KB
train.py 1KB
__pycache__
model.cpython-38.pyc 4KB
dataset.cpython-38.pyc 3KB
lightning_wrapper.cpython-38.pyc 4KB
README.md 991B
model_search.py 723B
共 21 条
- 1
资源评论
- 2301_774852052024-04-25这个资源总结的也太全面了吧,内容详实,对我帮助很大。
- m0_509816922024-04-16资源不错,很实用,内容全面,介绍详细,很好用,谢谢分享。
- 靓超2023-12-27感谢大佬,让我及时解决了当下的问题,解燃眉之急,必须支持!Make程序设计2024-01-24嗯嗯,谢谢,有问题随时私信沟通。
Make程序设计
- 粉丝: 5722
- 资源: 3570
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功