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

modify abbr postprocess

Browse Source
This commit is contained in:
lzr265946 2023-03-21 12:55:16 +08:00
parent 837c5001d4
commit 2ab8b7f473
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
funasr/utils/postprocess_utils.py
View File

@ -106,17 +106,18 @@ def abbr_dispose(words: List[Any], time_stamp: List[List] = None) -> List[Any]:
if num in abbr_begin: if num in abbr_begin:
if time_stamp is not None: if time_stamp is not None:
begin = time_stamp[ts_nums[num]][0] begin = time_stamp[ts_nums[num]][0]
word_lists.append(words[num].upper()) abbr_word = words[num].upper()
num += 1 num += 1
while num < words_size: while num < words_size:
if num in abbr_end: if num in abbr_end:
word_lists.append(words[num].upper()) abbr_word += words[num].upper()
last_num = num last_num = num
break break
else: else:
if words[num].encode('utf-8').isalpha(): if words[num].encode('utf-8').isalpha():
word_lists.append(words[num].upper()) abbr_word += words[num].upper()
num += 1 num += 1
word_lists.append(abbr_word)
if time_stamp is not None: if time_stamp is not None:
end = time_stamp[ts_nums[num]][1] end = time_stamp[ts_nums[num]][1]
ts_lists.append([begin, end]) ts_lists.append([begin, end])