增加README
This commit is contained in:
parent
9fe80b02a8
commit
2b45bd6d60
@ -8,10 +8,8 @@
|
||||
@time:2023/09/15
|
||||
"""
|
||||
from pathlib import Path
|
||||
|
||||
import numpy as np
|
||||
import pandas as pd
|
||||
|
||||
from utils.SignalPreprocess import XinXiaoPreprocess
|
||||
from utils.ModelDetection import SA_Detect
|
||||
from utils.ResultSummary import AnalyseSegment
|
||||
@ -72,7 +70,7 @@ def main(opt):
|
||||
if __name__ == '__main__':
|
||||
parser = argparse.ArgumentParser(description='SA Detection Configuration ----- Marques')
|
||||
parser.add_argument('-d', '--data_path', nargs='?', type=str, default='./Data', help='待测试数据文件路径或文件夹')
|
||||
parser.add_argument('--hz', type=int, nargs='?', default=1000, help='信号采样率')
|
||||
parser.add_argument('-n', '--hz', type=int, nargs='?', default=1000, help='信号采样率')
|
||||
parser.add_argument('-m', '--model', nargs='?', default=0, choices=[0, 1, 3, 4], help='选择一个模型路径')
|
||||
parser.add_argument('-b', "--batch_size", nargs="?", default=4096, type=int, help="模型每次预测片段数量")
|
||||
parser.add_argument('-o', '--output', nargs='?', default='./Output', help='输出文件夹')
|
||||
@ -80,9 +78,13 @@ if __name__ == '__main__':
|
||||
option = parser.parse_args()
|
||||
|
||||
# 手动输入,优先级高于默认值和命令行输入
|
||||
option.data_path = "/home/marques/code/marques/apnea/dataset/zhongda/zhongda_origin_npy/3103.npy"
|
||||
# option.batch_size = 1
|
||||
# option.TST = 565
|
||||
# option.data_path = "/home/marques/code/marques/apnea/dataset/zhongda/zhongda_origin_npy/3103.npy"
|
||||
# option.data_path = "/home/marques/code/marques/apnea/dataset/zhongda/zhongda_origin_npy"
|
||||
# option.hz = 1000
|
||||
# option.model = 0
|
||||
# option.batch_size = 1
|
||||
# option.output = "./Output"
|
||||
# option.TST = 565
|
||||
|
||||
|
||||
main(option)
|
||||
|
89
README.md
89
README.md
@ -1,2 +1,91 @@
|
||||
# 0915CXH_DL_SA
|
||||
|
||||
## 文件目录
|
||||
```bash
|
||||
0915CXH_DL_SA 主目录
|
||||
├── Data 数据
|
||||
│ ├── 3025.npy npy格式的数据
|
||||
│ ├── 3026.txt txt格式的数据
|
||||
│ ├── .......
|
||||
│ └── 3033.npy
|
||||
├── Output 存放暂停发生起止时间
|
||||
│ └── 3025.csv
|
||||
├── SADetectModel 存放模型结构文件和训练好的模型
|
||||
│ ├── SaDetectModel.py 模型结构文件
|
||||
│ └── SAmodel0.pt 模型权重文件
|
||||
├── utils 基本库
|
||||
│ ├── SignalPreprocess.py 心晓数据预处理
|
||||
│ ├── ModelDetection.py 模型检测
|
||||
│ └── ResultSummary.py 片段结果后处理
|
||||
├── ApneaDetection.py 启动入口
|
||||
└── README.md 项目说明
|
||||
|
||||
```
|
||||
|
||||
|
||||
## 开发环境
|
||||
实际运行环境可以不用严格按照下面的环境配置,只要能运行即可。
|
||||
- numpy==1.20.3
|
||||
- pandas==1.3.3
|
||||
- tqdm==4.64.0
|
||||
- scipy==1.7.3
|
||||
- torch==1.11.0+cu113
|
||||
- torchvision==0.12.0+cu113
|
||||
- torchaudio==0.11.0+cu113
|
||||
|
||||
## 使用方法
|
||||
|
||||
### 1. 从PyCharm中运行
|
||||
从0915CXH_DL_SA文件夹中打开PyCharm,
|
||||
打开**Main_Quality_Relabel**.py
|
||||
修改
|
||||
```bash
|
||||
option.data_path # 数据或文件夹路径
|
||||
option.hz # 采样频率 默认为1000Hz
|
||||
option.model # 模型选择 仅支持0、1、3、4
|
||||
option.batch_size # 批处理大小 默认为4096
|
||||
option.output # 输出文件夹路径
|
||||
option.TST # 真实睡眠时长,仅支持单文件输入,留空则以数据长度为真实睡眠时长
|
||||
```
|
||||
|
||||
|
||||
|
||||
|
||||
|
||||
### 2. 从命令行运行
|
||||
**注意:命令行传入变量优先级低于程序内设置的变量**
|
||||
从命令行中传入参数,无需打开pycharm,直接在代码文件夹的路径栏输入cmd后回车,进入到命令提示符
|
||||
,再在命令提示符中运行python 程序名 参数 即可运行
|
||||
|
||||
#### 支持参数
|
||||
|
||||
```bash
|
||||
usage: ApneaDetection.py [-h]
|
||||
[-d [DATA_PATH]]
|
||||
[-n [FREQUENCY]]
|
||||
[-m [MODEL_NUMBER]]
|
||||
[-b [BATCH_SIZE]]
|
||||
[-o [OUTPUT_PATH]]
|
||||
[-t [TST]]
|
||||
|
||||
|
||||
optional arguments:
|
||||
-h, --help
|
||||
显示帮助信息
|
||||
-d [DATA_PATH], --data_path [DATA_PATH]
|
||||
数据文件或文件夹路径
|
||||
-n [FREQUENCY], --frequency [FREQUENCY]
|
||||
采样频率 默认为1000Hz
|
||||
-m [MODEL_NUMBER], --model_number [MODEL_NUMBER]
|
||||
模型选择 仅支持0、1、3、4
|
||||
-b [BATCH_SIZE], --batch_size [BATCH_SIZE]
|
||||
批处理大小 默认为4096
|
||||
-o [OUTPUT_PATH], --output_path [OUTPUT_PATH]
|
||||
输出文件夹路径
|
||||
-t [TST], --TST [TST]
|
||||
真实睡眠时长,仅支持单文件输入,留空则以数据长度为真实睡眠时长
|
||||
|
||||
```
|
||||
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user