提取内容特征改为线程池

332plim · Jun 9, 2023 · c9367bb · c9367bb
1 parent a16c785
commit c9367bb
Showing 1 changed file with 8 additions and 8 deletions.
diff --git a/prepare/preprocess_ppg.py b/prepare/preprocess_ppg.py
@@ -67,15 +67,15 @@ def process_file(file):
             else:
                 process_num = args.thread_count
 
-            # with ThreadPoolExecutor(max_workers=process_num) as executor:
-            #     futures = [executor.submit(process_file, file) for file in os.listdir(f"./{wavPath}/{spks}")]
-            #     for future in tqdm(as_completed(futures), total=len(futures)):
-            #         pass
+            with ThreadPoolExecutor(max_workers=process_num) as executor:
+                futures = [executor.submit(process_file, file) for file in os.listdir(f"./{wavPath}/{spks}")]
+                for future in tqdm(as_completed(futures), total=len(futures)):
+                    pass
 
-            with Pool(processes=process_num) as pool:
-                results = [pool.apply_async(process_file, (file,)) for file in os.listdir(f"./{wavPath}/{spks}")]
-                for result in tqdm(results, total=len(results)):
-                    result.wait()
+            # with Pool(processes=process_num) as pool:
+            #     results = [pool.apply_async(process_file, (file,)) for file in os.listdir(f"./{wavPath}/{spks}")]
+            #     for result in tqdm(results, total=len(results)):
+            #         result.wait()
 
         else:
             file = spks