From f04f8a9e0339ddf955e8c9643d501d91edbaf069 Mon Sep 17 00:00:00 2001 From: marques <20172333133@m.scnu.edu.cn> Date: Thu, 21 Apr 2022 09:53:27 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=91=BD=E4=BB=A4=E8=A1=8C?= =?UTF-8?q?=E4=BA=8B=E4=BB=B6=E9=80=89=E6=8B=A9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- Main_Quality_Relabel.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) diff --git a/Main_Quality_Relabel.py b/Main_Quality_Relabel.py index f3d5dac..0191f79 100644 --- a/Main_Quality_Relabel.py +++ b/Main_Quality_Relabel.py @@ -29,7 +29,7 @@ all_path = [PSG_Data_Path, PSG_Label_Path, BCG_Data_Path, BCG_Label_Path] # 要遍历的事件 # 可选一个或多个 "Hypopnea" "Central apnea" "Obstructive apnea" "Mixed apnea" -focus_event_list = ["Obstructive apnea"] +focus_event_list = ["OSA"] # 信号显示事件前多少秒 front_add_second = 60 # 信号显示事件后多少秒 @@ -43,7 +43,8 @@ start_bcg_index = 0 if __name__ == '__main__': parser = argparse.ArgumentParser() parser.add_argument('-n', '--sampNo', nargs='?', type=int, default=sampNo, help='samp number') - parser.add_argument('-e', '--event', nargs='*', type=list, default='OSA', choices=['OSA', 'CSA', 'MAS', 'HPY'], + parser.add_argument('-e', '--event', nargs='*', type=str, default=focus_event_list, + choices=['OSA', 'CSA', 'MSA', 'HPY'], help='focus_event') parser.add_argument('-i', '--start_bcg_index', nargs='?', type=int, default=start_bcg_index, help='samp number') parser.add_argument('-f', '--front_add_second', nargs='?', type=int, default=front_add_second, @@ -52,6 +53,16 @@ if __name__ == '__main__': help='Add x second signal after the event') opt = parser.parse_args() + focus_event_list = [] + if 'OSA' in opt.focus_event_list: + focus_event_list.append("Obstructive apnea") + if 'CSA' in opt.focus_event_list: + focus_event_list.append("Central apnea") + if 'MSA' in opt.focus_event_list: + focus_event_list.append("Mixed apnea") + if 'HPY' in opt.focus_event_list: + focus_event_list.append("Hypopnea") + qualityRelabel = Quality_Relabel(all_path=all_path, sampNo=opt.sampNo, frequency=frequency, bcg_frequency=bcg_frequency, focus_event_list=focus_event_list)