From 848855c23db6ba0f22da0d513d79043afdca2b19 Mon Sep 17 00:00:00 2001 From: hypox64 Date: Fri, 3 May 2019 14:39:24 +0800 Subject: [PATCH] Updata readme --- README.md | 8 ++++---- how_to_run.md | 12 ++++++------ 2 files changed, 10 insertions(+), 10 deletions(-) diff --git a/README.md b/README.md index 0fb4696..38b6d65 100644 --- a/README.md +++ b/README.md @@ -1,5 +1,6 @@ # candock 这是一个用于记录毕业设计的日志仓库,其目的是尝试多种不同的深度神经网络结构(如LSTM,RESNET,DFCNN等)对单通道EEG进行自动化睡眠阶段分期.我们相信这些代码同时可以用于其他生理信号(如ECG,EMG等)的分类.希望这将有助于您的研究.
+![image](https://github.com/HypoX64/candock/blob/master/image/compare.png) ## 如何运行 如果你需要运行这些代码(训练自己的模型或者使用预训练模型进行测试)请进入以下页面
[How to run codes](https://github.com/HypoX64/candock/blob/master/how_to_run.md)
@@ -30,7 +31,7 @@ * multi_scale_resnet_1d 网络结构
该网络参考[geekfeiw / Multi-Scale-1D-ResNet](https://github.com/geekfeiw/Multi-Scale-1D-ResNet)
- 修改后的[网络结构如图](https://github.com/HypoX64/candock/blob/master/image/multi_scale_resnet_1d_network.png)
+ 修改后的[网络结构](https://github.com/HypoX64/candock/blob/master/image/multi_scale_resnet_1d_network.png)
* 关于交叉验证
为了便于与其他文献中的方法便于比较,使用了两种交叉验证方法
@@ -42,14 +43,13 @@ accuracy = (TP+TN)/(TP+FN+TN+FP)
recall = sensitivity = (TP)/(TP+FN)
对于总体:
- - Top1.err. + Top1.err.
* 关于代码
目前的代码仍然在不断修改与更新中,不能确保其能工作.详细内容将会在毕业设计完成后抽空更新.
## 部分实验结果 该部分将持续更新... ...
-[[Confusion matrix]](https://github.com/HypoX64/candock/blob/master/image/confusion_mat)
+[[Confusion matrix]](https://github.com/HypoX64/candock/blob/master/confusion_mat)
#### 5-Fold Cross-Validation Results * sleep-edf
diff --git a/how_to_run.md b/how_to_run.md index 52d90b7..e8ba28c 100644 --- a/how_to_run.md +++ b/how_to_run.md @@ -2,7 +2,7 @@ - Linux, Windows,mac - CPU or NVIDIA GPU + CUDA CuDNN - Python 3.5+ -- Pytroch 1.0+ +- Pytroch 1.0 ## Dependencies This code depends on torchvision, numpy, scipy, h5py, matplotlib, mne , requests, hashlib, available via pip install.
@@ -23,20 +23,20 @@ git clone https://github.com/HypoX64/candock cd candock ``` ### Train -* download datasets +* Download datasets ```bash python3 download_dataset.py ``` -* choose your options and run +* Input your options and run ```bash python3 train.py --dataset_dir './datasets/sleep-edfx/' --dataset_name sleep-edfx --signal_name 'EEG Fpz-Cz' --sample_num 10 --model_name lstm --batchsize 64 --network_save_freq 5 --epochs 20 --lr 0.0005 --BID 5_95_th --select_sleep_time --cross_validation subject ``` * Notes
-If want to use cpu to train or test, please use --no_cuda +If want to use cpu to train or test, please input --no_cuda ### Simple Test -* download pretrained model & simple test data [Google Drive](https://drive.google.com/open?id=1pup2_tZFGQQwB-hoXRjpMxiD4Vmpn0Lf) -* choose your options and run +* Download pretrained model & simple test data [[Google Drive]](https://drive.google.com/open?id=1pup2_tZFGQQwB-hoXRjpMxiD4Vmpn0Lf) +* Input your options and run ```bash python3 simple_test.py ``` \ No newline at end of file -- GitLab