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

准备构建数据集

Browse Source
This commit is contained in:
marques 2025-11-11 14:21:25 +08:00
parent 60e245b1e3
commit 1a0761c6c8
6 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
SHHS_process.py → dataset_builder/HYS_dataset.py
View File

0
dataset_builder/__init__.py Normal file
View File

10
HYS_process.py → event_mask_process/HYS_process.py
View File

@ -160,7 +160,7 @@ def process_one_signal(samp_id, show=False):
movement_list=resp_movement_position_list, movement_list=resp_movement_position_list,
sampling_rate=resp_fs, sampling_rate=resp_fs,
**resp_amp_change_conf, **resp_amp_change_conf,
verbose=True) verbose=False)
print( print(
f"amp_change_mask_shape: {resp_amp_change_mask.shape}, num_amp_change: {np.sum(resp_amp_change_mask == 1)}, count_amp_change_positions: {len(resp_amp_change_list)}") f"amp_change_mask_shape: {resp_amp_change_mask.shape}, num_amp_change: {np.sum(resp_amp_change_mask == 1)}, count_amp_change_positions: {len(resp_amp_change_list)}")
else: else:
@ -259,8 +259,8 @@ def process_one_signal(samp_id, show=False):
if __name__ == '__main__': if __name__ == '__main__':
yaml_path = Path("./dataset_config/HYS_config.yaml") yaml_path = Path("../dataset_config/HYS_config.yaml")
disable_df_path = Path("./排除区间.xlsx") disable_df_path = Path("../排除区间.xlsx")
conf = utils.load_dataset_conf(yaml_path) conf = utils.load_dataset_conf(yaml_path)
select_ids = conf["select_ids"] select_ids = conf["select_ids"]
@ -276,8 +276,8 @@ if __name__ == '__main__':
all_samp_disable_df = utils.read_disable_excel(disable_df_path) all_samp_disable_df = utils.read_disable_excel(disable_df_path)
process_one_signal(select_ids[9], show=True) process_one_signal(select_ids[6], show=True)
#
# for samp_id in select_ids: # for samp_id in select_ids:
# print(f"Processing sample ID: {samp_id}") # print(f"Processing sample ID: {samp_id}")
# process_one_signal(samp_id, show=False) # process_one_signal(samp_id, show=False)

0
event_mask_process/SHHS_process.py Normal file
View File

4
ZD5Y_process.py → event_mask_process/ZD5Y_process.py
View File

@ -252,8 +252,8 @@ def process_one_signal(samp_id, show=False):
if __name__ == '__main__': if __name__ == '__main__':
yaml_path = Path("./dataset_config/ZD5Y_config.yaml") yaml_path = Path("../dataset_config/ZD5Y_config.yaml")
disable_df_path = Path("./排除区间.xlsx") disable_df_path = Path("../排除区间.xlsx")
conf = utils.load_dataset_conf(yaml_path) conf = utils.load_dataset_conf(yaml_path)
select_ids = conf["select_ids"] select_ids = conf["select_ids"]

0
event_mask_process/__init__.py Normal file
View File