marques/DataPrepare
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

修正SA_Score的处理逻辑，移除不必要的Alpha转换，优化数据掩码生成

Browse Source
This commit is contained in:
marques 2025-11-17 09:40:43 +08:00
parent 19d476d489
commit d829f3e43d
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
draw_tools/draw_label.py
View File

@ -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') color='green', alpha=0.8, label='Amplitude Change Mask')
for event_code in event_codes: for event_code in event_codes:
sa_mask = event_mask["SA_Label"][segment_start:segment_end].repeat(signal_fs) == event_code 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 = (sa_mask * score_mask).astype(float)
y = (chn_signal[segment_start * signal_fs:segment_end * signal_fs] * sa_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) 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_"): if mask.startswith("Resp_") or mask.startswith("BCG_"):
event_mask[mask] = utils.none_to_nan_mask(event_mask[mask], 0) 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"] = utils.none_to_nan_mask(event_mask["SA_Score"], 0)
event_mask["SA_Score_Alpha"] = utils.none_to_nan_mask(event_mask["SA_Score_Alpha"], 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() fig, axes = create_psg_bcg_figure()
for segment_start, segment_end in tqdm(segment_list): 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_"): if mask.startswith("Resp_"):
resp_label[mask] = utils.none_to_nan_mask(resp_label[mask], 0) 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_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_Label_Alpha"] = utils.none_to_nan_mask(resp_label["Resp_Label_Alpha"], 0)
fig, axes = create_resp_figure() fig, axes = create_resp_figure()
for segment_start, segment_end in segment_list: for segment_start, segment_end in segment_list: