From d829f3e43d6ebc7d6e6aaa3ed215dedee23bb63a Mon Sep 17 00:00:00 2001 From: marques Date: Mon, 17 Nov 2025 09:40:43 +0800 Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3SA=5FScore=E7=9A=84=E5=A4=84?= =?UTF-8?q?=E7=90=86=E9=80=BB=E8=BE=91=EF=BC=8C=E7=A7=BB=E9=99=A4=E4=B8=8D?= =?UTF-8?q?=E5=BF=85=E8=A6=81=E7=9A=84Alpha=E8=BD=AC=E6=8D=A2=EF=BC=8C?= =?UTF-8?q?=E4=BC=98=E5=8C=96=E6=95=B0=E6=8D=AE=E6=8E=A9=E7=A0=81=E7=94=9F?= =?UTF-8?q?=E6=88=90?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- draw_tools/draw_label.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/draw_tools/draw_label.py b/draw_tools/draw_label.py index 134202e..6b0dcb9 100644 --- a/draw_tools/draw_label.py +++ b/draw_tools/draw_label.py @@ -125,7 +125,7 @@ def plt_signal_label_on_ax(ax: Axes, signal_data, segment_start, segment_end, ev color='green', alpha=0.8, label='Amplitude Change Mask') for event_code in event_codes: sa_mask = event_mask["SA_Label"][segment_start:segment_end].repeat(signal_fs) == event_code - score_mask = event_mask["SA_Score_Alpha"][segment_start:segment_end].repeat(signal_fs) + score_mask = event_mask["SA_Score"][segment_start:segment_end].repeat(signal_fs) # y = (sa_mask * score_mask).astype(float) y = (chn_signal[segment_start * signal_fs:segment_end * signal_fs] * sa_mask).astype(float) np.place(y, y == 0, np.nan) @@ -191,8 +191,10 @@ def draw_psg_bcg_label(psg_data, psg_label, bcg_data, event_mask, segment_list, if mask.startswith("Resp_") or mask.startswith("BCG_"): event_mask[mask] = utils.none_to_nan_mask(event_mask[mask], 0) - event_mask["SA_Score_Alpha"] = score_mask2alpha(event_mask["SA_Score"]) - event_mask["SA_Score_Alpha"] = utils.none_to_nan_mask(event_mask["SA_Score_Alpha"], 0) + event_mask["SA_Score"] = utils.none_to_nan_mask(event_mask["SA_Score"], 0) + + # event_mask["SA_Score_Alpha"] = score_mask2alpha(event_mask["SA_Score"]) + # event_mask["SA_Score_Alpha"] = utils.none_to_nan_mask(event_mask["SA_Score_Alpha"], 0) fig, axes = create_psg_bcg_figure() for segment_start, segment_end in tqdm(segment_list): @@ -227,8 +229,8 @@ def draw_resp_label(resp_data, resp_label, segment_list): if mask.startswith("Resp_"): resp_label[mask] = utils.none_to_nan_mask(resp_label[mask], 0) - resp_label["Resp_Score_Alpha"] = score_mask2alpha(resp_label["Resp_Score"]) - resp_label["Resp_Label_Alpha"] = utils.none_to_nan_mask(resp_label["Resp_Label_Alpha"], 0) + # resp_label["Resp_Score_Alpha"] = score_mask2alpha(resp_label["Resp_Score"]) + # resp_label["Resp_Label_Alpha"] = utils.none_to_nan_mask(resp_label["Resp_Label_Alpha"], 0) fig, axes = create_resp_figure() for segment_start, segment_end in segment_list: