From 2b45bd6d60296b2c0c939c7b48e149d617ba6abd Mon Sep 17 00:00:00 2001 From: andrew Date: Sun, 17 Sep 2023 09:52:31 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0README?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ApneaDetection.py | 14 ++++---- README.md | 89 +++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 97 insertions(+), 6 deletions(-) diff --git a/ApneaDetection.py b/ApneaDetection.py index 14ff905..f763433 100644 --- a/ApneaDetection.py +++ b/ApneaDetection.py @@ -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) diff --git a/README.md b/README.md index 5c14a16..0ab3346 100644 --- a/README.md +++ b/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] + 真实睡眠时长，仅支持单文件输入，留空则以数据长度为真实睡眠时长 + +``` + + +