From 38524b21d9bcbe6c0523a03b24d19326008a9b19 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B8=B8=E9=9B=81?= Date: Fri, 12 Jan 2024 13:29:37 +0800 Subject: [PATCH] funasr1.0 streaming --- .../industrial_data_pretraining/fsmn_vad_streaming/demo.py | 5 +++-- .../paraformer_streaming/demo.py | 6 +++--- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/examples/industrial_data_pretraining/fsmn_vad_streaming/demo.py b/examples/industrial_data_pretraining/fsmn_vad_streaming/demo.py index 6761a807a..6831cbae7 100644 --- a/examples/industrial_data_pretraining/fsmn_vad_streaming/demo.py +++ b/examples/industrial_data_pretraining/fsmn_vad_streaming/demo.py @@ -26,9 +26,10 @@ chunk_stride = int(chunk_size * 16000 / 1000) cache = {} -for i in range(int(len((speech)-1)/chunk_stride+1)): +total_chunk_num = int(len((speech)-1)/chunk_stride+1) +for i in range(total_chunk_num): speech_chunk = speech[i*chunk_stride:(i+1)*chunk_stride] - is_final = i == int(len((speech)-1)/chunk_stride+1) - 1 + is_final = i == total_chunk_num - 1 res = model(input=speech_chunk, cache=cache, is_final=is_final, diff --git a/examples/industrial_data_pretraining/paraformer_streaming/demo.py b/examples/industrial_data_pretraining/paraformer_streaming/demo.py index 872843b8e..65f182ca5 100644 --- a/examples/industrial_data_pretraining/paraformer_streaming/demo.py +++ b/examples/industrial_data_pretraining/paraformer_streaming/demo.py @@ -28,10 +28,10 @@ speech, sample_rate = soundfile.read(wav_file) chunk_stride = chunk_size[1] * 960 # 600ms、480ms cache = {} - -for i in range(int(len((speech)-1)/chunk_stride+1)): +total_chunk_num = int(len((speech)-1)/chunk_stride+1) +for i in range(total_chunk_num): speech_chunk = speech[i*chunk_stride:(i+1)*chunk_stride] - is_final = i == int(len((speech)-1)/chunk_stride+1) - 1 + is_final = i == total_chunk_num - 1 res = model(input=speech_chunk, cache=cache, is_final=is_final,