speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update timestamp for batch inference

Browse Source
This commit is contained in:
shixian.shi 2023-05-11 15:50:17 +08:00
parent 85f4aab3d3
commit b3bc53e332
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
funasr/bin/asr_inference_paraformer_vad_punc.py
View File

@ -642,7 +642,16 @@ def inference_modelscope(
# result = result_segments[0] # result = result_segments[0]
text, token, token_int = result[0], result[1], result[2] text, token, token_int = result[0], result[1], result[2]
time_stamp = None if len(result) < 5 else result[4] time_stamp = None if len(result) < 5 else result[4]
# post process timestamp
offset_list = [0]
offset = 0
for i in range(1, len(time_stamp)):
if time_stamp[i][0] < time_stamp[i-1][1]:
offset += time_stamp[i-1][1]
offset_list.append(offset)
for i in range(1, len(time_stamp)):
time_stamp[i][0] += offset_list[i]
time_stamp[i][1] += offset_list[i]
if use_timestamp and time_stamp is not None: if use_timestamp and time_stamp is not None:
postprocessed_result = postprocess_utils.sentence_postprocess(token, time_stamp) postprocessed_result = postprocess_utils.sentence_postprocess(token, time_stamp)