49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: UTF-8 -*-
|
|
"""
|
|
@author:Marques
|
|
@file:Main_Quality_Relabel.py
|
|
@email:admin@marques22.com
|
|
@email:2021022362@m.scnu.edu.cn
|
|
@time:2022/03/28
|
|
"""
|
|
from pathlib import Path
|
|
from utils.Quality_Relabel import Quality_Relabel
|
|
|
|
# start-----一般不用修改------
|
|
# 绘图时的采样率
|
|
frequency = 100
|
|
# 心晓数据采样率
|
|
bcg_frequency = 1000
|
|
# PSG_Data_Path = Path("Data/PSG/")
|
|
# PSG_Label_Path = Path("Data/PSG_label/")
|
|
# BCG_Data_Path = Path("Data/BCG/")
|
|
# BCG_Label_Path = Path("Data/BCG_label/")
|
|
PSG_Data_Path = Path("I:/CXH/Quality_Relabel/Data/PSG/")
|
|
PSG_Label_Path = Path("I:/CXH/Quality_Relabel/Data/PSG_label/")
|
|
BCG_Data_Path = Path("I:/CXH/Quality_Relabel/Data/BCG/")
|
|
BCG_Label_Path = Path("I:/CXH/Quality_Relabel/Data/BCG_label/")
|
|
all_path = [PSG_Data_Path, PSG_Label_Path, BCG_Data_Path, BCG_Label_Path]
|
|
# end-----一般不用修改------
|
|
|
|
# 要遍历的事件
|
|
# 可选一个或多个 "Hypopnea" "Central apnea" "Obstructive apnea" "Mixed apnea"
|
|
focus_event_list = ["Obstructive apnea"]
|
|
# 信号显示事件前多少秒
|
|
front_add_second = 60
|
|
# 信号显示事件后多少秒
|
|
back_add_second = 60
|
|
|
|
# 样本编号
|
|
sampNo = 670
|
|
# 从第几个心晓事件数量开始
|
|
start_bcg_index = 0
|
|
|
|
if __name__ == '__main__':
|
|
qualityRelabel = Quality_Relabel(all_path=all_path, sampNo=sampNo, frequency=frequency, bcg_frequency=bcg_frequency,
|
|
focus_event_list=focus_event_list)
|
|
|
|
qualityRelabel.show_all_event(start_bcg_index=start_bcg_index,
|
|
front_add_second=front_add_second,
|
|
back_add_second=back_add_second)
|