增加README

This commit is contained in:
andrew 2023-09-17 09:52:31 +08:00
parent 9fe80b02a8
commit 2b45bd6d60
2 changed files with 97 additions and 6 deletions

View File

@ -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)

View File

@ -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]
真实睡眠时长,仅支持单文件输入,留空则以数据长度为真实睡眠时长
```