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

fix uniasr postprocess

Browse Source
This commit is contained in:
lzr265946 2023-03-24 10:52:59 +08:00
parent 78a3b75cc0
commit da830cb825
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
funasr/bin/asr_inference_uniasr.py
View File

@ -261,6 +261,7 @@ class Speech2Text:
# Change integer-ids to tokens # Change integer-ids to tokens
token = self.converter.ids2tokens(token_int) token = self.converter.ids2tokens(token_int)
token = list(filter(lambda x: x != "<gbg>", token))
if self.tokenizer is not None: if self.tokenizer is not None:
text = self.tokenizer.tokens2text(token) text = self.tokenizer.tokens2text(token)
@ -512,7 +513,7 @@ def inference_modelscope(
finish_count += 1 finish_count += 1
asr_utils.print_progress(finish_count / file_count) asr_utils.print_progress(finish_count / file_count)
if writer is not None: if writer is not None:
ibest_writer["text"][key] = text ibest_writer["text"][key] = text_postprocessed
return asr_result_list return asr_result_list
return _forward return _forward

3
funasr/bin/asr_inference_uniasr_vad.py
View File

@ -261,6 +261,7 @@ class Speech2Text:
# Change integer-ids to tokens # Change integer-ids to tokens
token = self.converter.ids2tokens(token_int) token = self.converter.ids2tokens(token_int)
token = list(filter(lambda x: x != "<gbg>", token))
if self.tokenizer is not None: if self.tokenizer is not None:
text = self.tokenizer.tokens2text(token) text = self.tokenizer.tokens2text(token)
@ -512,7 +513,7 @@ def inference_modelscope(
finish_count += 1 finish_count += 1
asr_utils.print_progress(finish_count / file_count) asr_utils.print_progress(finish_count / file_count)
if writer is not None: if writer is not None:
ibest_writer["text"][key] = text ibest_writer["text"][key] = text_postprocessed
return asr_result_list return asr_result_list
return _forward return _forward