Yorusora/Signal_Label_Reborn
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修改了文件命名规则

Browse Source
This commit is contained in:
Yorusora
2025-05-19 17:21:27 +08:00
parent 3511b94c28
commit 55d5df3fb6
4 changed files with 140 additions and 76 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
func/Module_SA_label.py Normal file
View File

@ -0,0 +1,32 @@
from gc import collect
from pathlib import Path
from traceback import format_exc
import matplotlib.pyplot as plt
from PySide6.QtWidgets import QMessageBox, QMainWindow, QApplication
from matplotlib import gridspec
from matplotlib.backends.backend_qt import NavigationToolbar2QT
from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg
from overrides import overrides
from pandas import read_csv, DataFrame
from scipy.signal import resample
from yaml import dump, load, FullLoader
from func.utils.PublicFunc import PublicFunc
from func.utils.Constants import Constants, ConfigParams
from func.utils.Result import Result
from ui.MainWindow.MainWindow_SA_label import Ui_MainWindow_SA_label
# from ui.setting.preprocess_input_setting import Ui_MainWindow_preprocess_input_setting
Config = {
}
class MainWindow_SA_label(QMainWindow):
def __init__(self):
super(MainWindow_SA_label, self).__init__()
self.ui = Ui_MainWindow_SA_label()
self.ui.setupUi(self)

37
func/Module_mainwindow.py
View File

@ -1,6 +1,7 @@
from pathlib import Path
from PySide6.QtWidgets import QMainWindow, QMessageBox, QFileDialog
from PySide6.QtGui import QGuiApplication
from matplotlib import use
from yaml import dump, load, FullLoader
@ -15,6 +16,7 @@ from func.Module_label_check import MainWindow_label_check
from func.Module_precisely_align import MainWindow_precisely_align
from func.Module_cut_PSG import MainWindow_cut_PSG
from func.Module_artifact_label import MainWindow_artifact_label
from func.Module_SA_label import MainWindow_SA_label
from func.utils.Constants import Constants, ConfigParams
@ -34,7 +36,15 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
self.ui = Ui_Signal_Label()
self.ui.setupUi(self)
self.setFixedSize(720, 1080) # 设置固定大小，禁止缩放
# self.setFixedSize(720, 1080) # 设置固定大小，禁止缩放
# 获得屏幕分辨率，以3：4比例设置窗口大小，
screen = QGuiApplication.primaryScreen()
size = screen.availableGeometry()
screen_height = size.height()
window_height = int(screen_height * 0.75)
window_width = int(window_height * 1 / 2)
self.resize(window_width, window_height)
# 消息弹窗初始化
self.msgBox = QMessageBox()
@ -53,6 +63,8 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
self.cut_PSG = None
self.artifact_label = None
self.SA_label = None
# 绑定槽函数
self.ui.pushButton_open.clicked.connect(self.__slot_btn_open__)
self.ui.pushButton_approximately_align.clicked.connect(self.__slot_btn_approximately_align__)
@ -65,6 +77,7 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
self.ui.pushButton_precisely_align.clicked.connect(self.__slot_btn_precisely_align__)
self.ui.pushButton_cut_PSG.clicked.connect(self.__slot_btn_cut_PSG__)
self.ui.pushButton_artifact_label.clicked.connect(self.__slot_btn_artifact_label__)
self.ui.pushButton_SA_label.clicked.connect(self.__slot_btn_SA_label__)
@staticmethod
def __read_config__():
@ -105,6 +118,8 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
if not self.check_sampID():
return
self.approximately_align.show(root_path, int(sampID))
# 默认最大化显示而非固定分辨率
self.approximately_align.showMaximized()
def __slot_btn_preprocess__(self):
self.preprocess = MainWindow_preprocess()
@ -123,6 +138,8 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
if not self.check_sampID():
return
self.preprocess.show(mode, root_path, int(sampID))
# 默认最大化显示而非固定分辨率
self.preprocess.showMaximized()
def __slot_btn_detect_Jpeak__(self):
self.detect_Jpeak = MainWindow_detect_Jpeak()
@ -133,6 +150,8 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
if not self.check_sampID():
return
self.detect_Jpeak.show(root_path, int(sampID))
# 默认最大化显示而非固定分辨率
self.detect_Jpeak.showMaximized()
def __slot_btn_detect_Rpeak__(self):
self.detect_Rpeak = MainWindow_detect_Rpeak()
@ -143,6 +162,8 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
if not self.check_sampID():
return
self.detect_Rpeak.show(root_path, int(sampID))
# 默认最大化显示而非固定分辨率
self.detect_Rpeak.showMaximized()
def __slot_btn_label_check__(self):
self.label_check = MainWindow_label_check()
@ -163,6 +184,9 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
else:
raise ValueError("模式不存在")
self.label_check.show(mode, root_path, int(sampID))
# 默认最大化显示而非固定分辨率
self.label_check.showMaximized()
def __slot_btn_precisely_align__(self):
self.precisely_align = MainWindow_precisely_align()
@ -173,6 +197,8 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
if not self.check_sampID():
return
self.precisely_align.show(root_path, int(sampID))
# 默认最大化显示而非固定分辨率
self.precisely_align.showMaximized()
def __slot_btn_cut_PSG__(self):
self.cut_PSG = MainWindow_cut_PSG()
@ -183,6 +209,8 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
if not self.check_sampID():
return
self.cut_PSG.show(root_path, int(sampID))
# 默认最大化显示而非固定分辨率
self.cut_PSG.showMaximized()
def __slot_btn_artifact_label__(self):
self.artifact_label = MainWindow_artifact_label()
@ -193,6 +221,12 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
if not self.check_sampID():
return
self.artifact_label.show(root_path, int(sampID))
# 默认最大化显示而非固定分辨率
self.artifact_label.showMaximized()
def __slot_btn_SA_label__(self):
self.SA_label = MainWindow_SA_label()
self.SA_label.show()
def seek_sampID(self, path):
if not Path(path).exists():
@ -202,6 +236,7 @@ class MainWindow(QMainWindow, Ui_Signal_Label):
sub_folders = [item.name for item in Path(path).iterdir() if item.is_dir()]
self.ui.comboBox_sampID.addItems(sub_folders)
def check_root_path(self):
if self.ui.plainTextEdit_root_path.toPlainText() == Constants.STRING_IS_EMPTY:
PublicFunc.msgbox_output(self, Constants.MAINWINDOW_ROOT_PATH_NOT_EXIST, Constants.MSGBOX_TYPE_ERROR)

8
func/Module_precisely_align.py
View File

@ -1642,10 +1642,10 @@ class Data:
orgfs = Config["orgfs"]
off = Config["offset_anchor"]
self.res_orgBcg = self.raw_orgBcg
self.res_BCG = self.raw_BCG
self.cut_ECG = self.raw_ECG
self.cut_Rpeak = self.Rpeak
self.res_orgBcg = self.raw_orgBcg.copy()
self.res_BCG = self.raw_BCG.copy()
self.cut_ECG = self.raw_ECG.copy()
self.cut_Rpeak = self.Rpeak.copy()
if off > 0:
self.cut_ECG = self.cut_ECG[off:]

139
数据结构化输入和输出命名规范.md
View File

@ -6,49 +6,50 @@
```
.../Label/<sampID>
|-Artifact_a.txt
|-Artifact_b.txt
|-Artifact_c.csv
|-Artifact_a_采样率.txt
|-Artifact_b_采样率.txt
|-Artifact_c_采样率.csv
|-SQ_label_10s.csv
|-SQ_label_30s.csv
|-Resp_quality_label.txt
|-Tho_peak.txt
|-Tho_peak_采样率.txt
|-SA Label_corrected.csv
|-SA Label_add.csv`
.../OrgBCG_Aligned/<sampID>
|-Align_info.yaml
|-BCG_Sync_采样率.txt
|-orgBcg_Sync_采样率.txt
|-Jpeak_Sync.txt
.../OrgBCG_Text/<sampID>
|-orgBcg_Raw_采样率.txt
|-BCG_Raw_采样率.txt
|-JPeak_revise.txt
|-JPeak_revise_corrected.txt
|-SA Label_add.csv
|-Precisely_Align_info.txt
|-Approximately_Align_Info.csv
.../OrgBCG_Aligned/<sampID>
|-BCG_Sync_采样率.txt
|-OrgBCG_Sync_采样率.txt
|-Jpeak_Sync_采样率.txt
.../OrgBCG_Text/<sampID>
|-OrgBCG_Raw_采样率.txt
|-BCG_Filter_采样率.txt
|-JPeak_revise_采样率.txt
|-JPeak_revise_corrected_采样率.txt
.../OrgBCG_Origin/<sampID>
|-...
.../PSG_Aligned/<sampID>
|-Align_info.yaml
|-ECG_Sync_采样率.txt
|-Rpeak_Sync.txt
|-Rpeak_Sync_采样率.txt
|-5_class_Sync_1.txt
|-SA Label_Sync.csv
|-Effort Abd_Sync_采样率.txt
|-Effort Tho_Sync_采样率.txt
|-Flow Patient_Sync_采样率.txt
|-Flow T_Sync_采样率.txt
|-Flow P_Sync_采样率.txt
|-Snore_Sync_采样率.txt
|-SpO2_Sync_采样率.txt
.../PSG_Text/<sampID>
|-orgEcg II_Raw_采样率.txt
|-ECG_Raw_采样率.txt
|-Rpeak_final.txt
|-Rpeak_final_corrected.txt
|-5_class_Raw_1.txt
|-ECG II_Raw_采样率.txt
|-ECG_Filter_采样率.txt
|-Rpeak_final_采样率.txt
|-Rpeak_final_corrected_采样率.txt
|-5_class_Raw_采样率.txt
|-SA Label_Raw.csv
|-Effort Abd_Raw_采样率.txt
|-Effort Tho_Raw_采样率.txt
|-Flow Patient_Raw_采样率.txt
|-Flow T_Raw_采样率.txt
|-Flow P_Raw_采样率.txt
|-Snore_Raw_采样率.txt
|-SpO2_Raw_采样率.txt
|-StartTime_Raw.txt
@ -66,13 +67,13 @@
输入：
原始orgBcg信号：`./OrgBCG_Text/<sampID>/orgBcg_Raw_采样率.txt`
原始orgBcg信号：`./OrgBCG_Text/<sampID>/OrgBCG_Raw_采样率.txt`
原始PSG信号：`./PSG_Text/<sampID>/Axxxxxxx.edf`
输出：
粗同步后的位置索引：`./OrgBCG_Text/<sampID>/Approximately_Align_Info.csv`
粗同步后的位置索引：`./Label/<sampID>/Approximately_Align_Info.csv`
### 2 预处理
@ -80,15 +81,15 @@
输入：
原始orgBcg信号：`./OrgBCG_Text/<sampID>/orgBcg_Raw_采样率.txt`
原始orgBcg信号：`./OrgBCG_Text/<sampID>/OrgBCG_Raw_采样率.txt`
原始ECG信号：`./PSG_Text/<sampID>/ECG II_Raw_采样率.txt`
输出：
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Raw_采样率.txt`
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Filter_采样率.txt`
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Raw_采样率.txt`
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Filter_采样率.txt`
### 3 数据精同步
@ -101,11 +102,11 @@
输入：
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Raw_采样率.txt`
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Filter_采样率.txt`
输出：
算法定位的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final.txt`
算法定位的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final_采样率.txt`
#### 3.1.2 J峰算法定位
@ -113,11 +114,11 @@
输入：
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Raw_采样率.txt`
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Filter_采样率.txt`
输出：
算法定位的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise.txt`
算法定位的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise_采样率.txt`
#### 3.2 人工纠正
@ -128,15 +129,15 @@
输入：
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Raw_采样率.txt`
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Filter_采样率.txt`
算法定位的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final.txt`
算法定位的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final_采样率.txt`
粗同步后的位置索引：`./OrgBCG_Text/<sampID>/Approximately_Align_Info.csv`
粗同步后的位置索引：`./Label/<sampID>/Approximately_Align_Info.csv`
输出：
人工纠正后的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final_corrected.txt`
人工纠正后的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final_corrected_采样率.txt`
#### 3.2.2 J峰人工纠正*
@ -144,15 +145,15 @@
输入：
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Raw_采样率.txt`
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Filter_采样率.txt`
算法定位的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise.txt`
算法定位的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise_采样率.txt`
粗同步后的位置索引：`./OrgBCG_Text/<sampID>/Approximately_Align_Info.csv`
粗同步后的位置索引：`./Label/<sampID>/Approximately_Align_Info.csv`
输出：
人工纠正后的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise_corrected.txt`
人工纠正后的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise_corrected_采样率.txt`
#### 3.3 数据片段起止对齐、数据采样率同步
@ -160,33 +161,31 @@
输入：
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Raw_采样率.txt`
滤波后的ECG信号：`./PSG_Text/<sampID>/ECG_Filter_采样率.txt`
人工纠正后的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final_corrected.txt`
人工纠正后的R峰坐标：`./PSG_Text/<sampID>/Rpeak_final_corrected_采样率.txt`
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Raw_采样率.txt`
带通滤波BCG信号：`./OrgBCG_Text/<sampID>/BCG_Filter_采样率.txt`
人工纠正后的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise_corrected.txt`
人工纠正后的J峰坐标：`./OrgBCG_Text/<sampID>/JPeak_revise_corrected_采样率.txt`
原始orgBcg信号：`./OrgBCG_Text/<sampID>/orgBcg_Raw_采样率.txt`
原始orgBcg信号：`./OrgBCG_Text/<sampID>/OrgBCG_Raw_采样率.txt`
粗同步后的位置索引：`./OrgBCG_Text/<sampID>/Approximately_Align_Info.csv`
粗同步后的位置索引：`./Label/<sampID>/Approximately_Align_Info.csv`
输出：
PSG的对齐信息：`./PSG_Aligned/<sampID>/Align_info.yaml`
精同步对齐信息：`./Label/<sampID>/Precisely_Align_info.txt`
同步后的ECG信号：`./PSG_Aligned/<sampID>/ECG_Sync_采样率.txt`
同步后的R峰坐标：`./PSG_Aligned/<sampID>/Rpeak_Sync.txt`
BCG的对齐信息：`./OrgBCG_Aligned/<sampID>/Align_info.yaml`
同步后的R峰坐标：`./PSG_Aligned/<sampID>/Rpeak_Sync_采样率.txt`
同步后的BCG信号：`./OrgBCG_Aligned/<sampID>/BCG_Sync_采样率.txt`
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/orgBcg_Sync_采样率.txt`
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/OrgBCG_Sync_采样率.txt`
同步后的J峰坐标：`./OrgBCG_Aligned/<sampID>/Jpeak_Sync.txt`
同步后的J峰坐标：`./OrgBCG_Aligned/<sampID>/Jpeak_Sync_采样率.txt`
#### 3.4、冗余数据切割、标签映射
@ -194,21 +193,19 @@ BCG的对齐信息：`./OrgBCG_Aligned/<sampID>/Align_info.yaml`
输入：
PSG的对齐信息：`./PSG_Aligned/<sampID>/Align_info.yaml`
精同步对齐信息：`./Label/<sampID>/Align_info.txt`
BCG的对齐信息：`./OrgBCG_Aligned/<sampID>/Align_info.yaml`
原始的其他PSG通道信号：`./PSG_Text/<sampID>/通道名_Raw_采样率.txt`（通道名包括：Effort Abd, Effort Tho, Flow T, Flow P, Snore, SpO2）
原始的其他PSG通道信号：`./PSG_Text/<sampID>/通道名_Raw_采样率.txt`（通道名包括：Effort Abd, Effort Tho, Flow Patient, Snore, SpO2）
原始的睡眠分期标签：`./PSG_Text/<sampID>/5_class_Raw_1.txt`
原始的睡眠分期标签：`./PSG_Text/<sampID>/5_class_Raw_采样率.txt`
原始的睡眠呼吸暂停事件标签：`./PSG_Text/<sampID>/SA Label_Raw.csv`
输出：
同步后的其他PSG通道信号：`./PSG_Aligned/<sampID>/通道名_Sync_采样率.txt`（通道名包括：Effort Abd, Effort Tho, Flow Patient, Snore, SpO2）
同步后的其他PSG通道信号：`./PSG_Aligned/<sampID>/通道名_Sync_采样率.txt`（通道名包括：Effort Abd, Effort Tho, Flow T, Flow P, Snore, SpO2）
同步后的睡眠分期标签：`./PSG_Aligned/<sampID>/5_class_Sync_1.txt`
同步后的睡眠分期标签：`./PSG_Aligned/<sampID>/5_class_Sync_采样率.txt`
同步后的睡眠呼吸暂停事件标签：`./PSG_Aligned/<sampID>/SA Label_Sync.csv`
@ -220,15 +217,15 @@ BCG的对齐信息：`./OrgBCG_Aligned/<sampID>/Align_info.yaml`
同步后的BCG信号：`./OrgBCG_Aligned/<sampID>/BCG_Sync_采样率.txt`
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/orgBcg_Sync_采样率.txt`
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/OrgBCG_Sync_采样率.txt`
输出：
txt格式的体动标签：`./Label/<sampID>/Artifact_a.txt`
txt格式的体动标签：`./Label/<sampID>/Artifact_a_采样率.txt`
体动标签类型数量统计：`./Label/<sampID>/Artifact_b.txt`
体动标签类型数量统计：`./Label/<sampID>/Artifact_b_采样率.txt`
csv格式的体动标签：`./Label/<sampID>/Artifact_c.csv`
csv格式的体动标签：`./Label/<sampID>/Artifact_c_采样率.csv`
### 5 质量评估
@ -238,7 +235,7 @@ csv格式的体动标签：`./Label/<sampID>/Artifact_c.csv`
同步后的BCG信号：`./OrgBCG_Aligned/<sampID>/BCG_Sync_采样率.txt`
txt格式的体动标签：`./Label/<sampID>/Artifact_a.txt`
txt格式的体动标签：`./Label/<sampID>/Artifact_a_采样率.txt`
输出：
@ -266,13 +263,13 @@ txt格式的体动标签：`./Label/<sampID>/Artifact_a.txt`
同步后的THO信号：`./PSG_Aligned/<sampID>/Effort Tho_Sync_采样率.txt`
txt格式的体动标签：`./Label/<sampID>/Artifact_a.txt`
txt格式的体动标签：`./Label/<sampID>/Artifact_a_采样率.txt`
输出：
呼吸可用性标签：`./Label/<sampID>/Resp_quality_label.txt`
THO信号呼吸间期标签：`./Label/<sampID>/Tho_peak.txt`
THO信号呼吸间期标签：`./Label/<sampID>/Tho_peak_采样率.txt`
### 8 呼吸暂停事件标注
@ -280,13 +277,13 @@ THO信号呼吸间期标签：`./Label/<sampID>/Tho_peak.txt`
输入：
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/orgBcg_Sync_采样率.txt`
同步后的orgBcg信号：`./OrgBCG_Aligned/<sampID>/OrgBCG_Sync_采样率.txt`
txt格式的体动标签：`./Label/<sampID>/Artifact_a.txt`
txt格式的体动标签：`./Label/<sampID>/Artifact_a_采样率.txt`
同步后的FlowT信号：`./PSG_Aligned/<sampID>/Effort Tho_Sync_采样率.txt`
同步后的Flow T信号：`./PSG_Aligned/<sampID>/Flow T_Sync_采样率.txt`
同步后的FlowP信号：`./PSG_Aligned/<sampID>/Effort Tho_Sync_采样率.txt`
同步后的Flow P信号：`./PSG_Aligned/<sampID>/Flow P_Sync_采样率.txt`
同步后的THO信号：`./PSG_Aligned/<sampID>/Effort Tho_Sync_采样率.txt`