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

Merge pull request #39 from alibaba-damo-academy/dev_lyb

Browse Source 
fix ctc beamsearch
This commit is contained in:
zhifu gao 2023-01-18 15:36:25 +08:00 committed by GitHub
commit 944c04721c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/bin/asr_inference_paraformer.py
View File

@ -169,7 +169,7 @@ class Speech2Text:
self.converter = converter
self.tokenizer = tokenizer
is_use_lm = lm_weight != 0.0 and lm_file is not None
if ctc_weight == 0.0 and not is_use_lm:
if (ctc_weight == 0.0 or asr_model.ctc == None) and not is_use_lm:
beam_search = None
self.beam_search = beam_search
logging.info(f"Beam_search: {self.beam_search}")

2
funasr/bin/asr_inference_paraformer_timestamp.py
View File

@ -172,7 +172,7 @@ class Speech2Text:
self.converter = converter
self.tokenizer = tokenizer
is_use_lm = lm_weight != 0.0 and lm_file is not None
if ctc_weight == 0.0 and not is_use_lm:
if (ctc_weight == 0.0 or asr_model.ctc == None) and not is_use_lm:
beam_search = None
self.beam_search = beam_search
logging.info(f"Beam_search: {self.beam_search}")

4
funasr/bin/asr_inference_paraformer_vad_punc.py
View File

@ -174,7 +174,7 @@ class Speech2Text:
self.converter = converter
self.tokenizer = tokenizer
is_use_lm = lm_weight != 0.0 and lm_file is not None
if ctc_weight == 0.0 and not is_use_lm:
if (ctc_weight == 0.0 or asr_model.ctc == None) and not is_use_lm:
beam_search = None
self.beam_search = beam_search
logging.info(f"Beam_search: {self.beam_search}")
@ -666,7 +666,7 @@ def inference_modelscope(
time_stamp_postprocessed))
logging.info("decoding, feature length total: {}, forward_time total: {:.4f}, rtf avg: {:.4f}".
format(length_total, forward_time_total, 100 * forward_time_total / (length_total * lfr_factor)))
format(length_total, forward_time_total, 100 * forward_time_total / (length_total * lfr_factor+1e-6)))
return asr_result_list
return _forward