From ff0b37e7d912f5bcf6eca6c0aabc6b6cbbc09752 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=E6=82=A6?= Date: Sun, 21 Jan 2024 13:02:12 +0800 Subject: [PATCH] =?UTF-8?q?=E8=AE=B2=E8=BD=AC=E5=86=99=E9=80=BB=E8=BE=91?= =?UTF-8?q?=E6=94=B9=E4=B8=BA=E5=A4=9A=E8=BF=9B=E7=A8=8B=E5=88=A9=E7=94=A8?= =?UTF-8?q?=E5=A4=9A=E6=A0=B8cpu=E6=8F=90=E9=AB=98=E8=BD=AC=E5=86=99?= =?UTF-8?q?=E6=95=88=E7=8E=87?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 讲转写逻辑改为多进程利用多核cpu提高转写效率 --- tools/damo_asr/cmd-asr.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/tools/damo_asr/cmd-asr.py b/tools/damo_asr/cmd-asr.py index 61227c36..22b11f78 100644 --- a/tools/damo_asr/cmd-asr.py +++ b/tools/damo_asr/cmd-asr.py @@ -30,7 +30,6 @@ inference_pipeline = pipeline( punc_model='tools/damo_asr/models/punc_ct-transformer_zh-cn-common-vocab272727-pytorch', ) - def process_audio_file(dir,name,opt_name): try: @@ -42,7 +41,6 @@ def process_audio_file(dir,name,opt_name): except: print(traceback.format_exc()) - def run__process(): # 主进程 opt_dir="output/asr_opt" @@ -53,7 +51,6 @@ def run__process(): # 主进程 with multiprocessing.Pool(processes=processes) as pool: pool.starmap(process_audio_file, [(dir, name ,opt_name) for name in os.listdir(dir)]) - if __name__ == '__main__': run__process()