准备构建数据集

event_mask_process/HYS_process.py

@@ -160,7 +160,7 @@ def process_one_signal(samp_id, show=False):
             movement_list=resp_movement_position_list,
             sampling_rate=resp_fs,
             **resp_amp_change_conf,
-            verbose=True)
+            verbose=False)
         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)}")
     else:
@@ -259,8 +259,8 @@ def process_one_signal(samp_id, show=False):
 
 
 if __name__ == '__main__':
-    yaml_path = Path("./dataset_config/HYS_config.yaml")
-    disable_df_path = Path("./排除区间.xlsx")
+    yaml_path = Path("../dataset_config/HYS_config.yaml")
+    disable_df_path = Path("../排除区间.xlsx")
 
     conf = utils.load_dataset_conf(yaml_path)
     select_ids = conf["select_ids"]
@@ -276,8 +276,8 @@ if __name__ == '__main__':
 
     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:
     #     print(f"Processing sample ID: {samp_id}")
     #     process_one_signal(samp_id, show=False)

event_mask_process/ZD5Y_process.py

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