EngineX-Iluvatar/enginex-mr_series-sherpa-onnx
Archived
16
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

feat: addition of blank_penalty config in online_recognizer (#1232)

Browse Source
This commit is contained in:
Parth Khiera
2024-08-08 06:40:17 +05:30
committed by GitHub
parent 8a5f5c1999
commit ba4cb6169f
3 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/go/sherpa_onnx.go
View File

@@ -129,6 +129,7 @@ type OnlineRecognizerConfig struct {
Rule3MinUtteranceLength float32 Rule3MinUtteranceLength float32
HotwordsFile string HotwordsFile string
HotwordsScore float32 HotwordsScore float32
BlankPenalty float32
CtcFstDecoderConfig OnlineCtcFstDecoderConfig CtcFstDecoderConfig OnlineCtcFstDecoderConfig
RuleFsts string RuleFsts string
RuleFars string RuleFars string
@@ -212,6 +213,7 @@ func NewOnlineRecognizer(config *OnlineRecognizerConfig) *OnlineRecognizer {
defer C.free(unsafe.Pointer(c.hotwords_file)) defer C.free(unsafe.Pointer(c.hotwords_file))
c.hotwords_score = C.float(config.HotwordsScore) c.hotwords_score = C.float(config.HotwordsScore)
c.blank_penalty = C.float(config.BlankPenalty)
c.rule_fsts = C.CString(config.RuleFsts) c.rule_fsts = C.CString(config.RuleFsts)
defer C.free(unsafe.Pointer(c.rule_fsts)) defer C.free(unsafe.Pointer(c.rule_fsts))
@@ -421,6 +423,7 @@ type OfflineRecognizerConfig struct {
MaxActivePaths int MaxActivePaths int
HotwordsFile string HotwordsFile string
HotwordsScore float32 HotwordsScore float32
BlankPenalty float32
RuleFsts string RuleFsts string
RuleFars string RuleFars string
} }
@@ -530,6 +533,8 @@ func NewOfflineRecognizer(config *OfflineRecognizerConfig) *OfflineRecognizer {
c.hotwords_score = C.float(config.HotwordsScore) c.hotwords_score = C.float(config.HotwordsScore)
c.blank_penalty = C.float(config.BlankPenalty)
c.rule_fsts = C.CString(config.RuleFsts) c.rule_fsts = C.CString(config.RuleFsts)
defer C.free(unsafe.Pointer(c.rule_fsts)) defer C.free(unsafe.Pointer(c.rule_fsts))

2
sherpa-onnx/c-api/c-api.cc
View File

@@ -105,6 +105,8 @@ SherpaOnnxOnlineRecognizer *SherpaOnnxCreateOnlineRecognizer(
recognizer_config.hotwords_score = recognizer_config.hotwords_score =
SHERPA_ONNX_OR(config->hotwords_score, 1.5); SHERPA_ONNX_OR(config->hotwords_score, 1.5);
recognizer_config.blank_penalty = SHERPA_ONNX_OR(config->blank_penalty, 0.0);
recognizer_config.ctc_fst_decoder_config.graph = recognizer_config.ctc_fst_decoder_config.graph =
SHERPA_ONNX_OR(config->ctc_fst_decoder_config.graph, ""); SHERPA_ONNX_OR(config->ctc_fst_decoder_config.graph, "");
recognizer_config.ctc_fst_decoder_config.max_active = recognizer_config.ctc_fst_decoder_config.max_active =

1
sherpa-onnx/c-api/c-api.h
View File

@@ -142,6 +142,7 @@ SHERPA_ONNX_API typedef struct SherpaOnnxOnlineRecognizerConfig {
/// Bonus score for each token in hotwords. /// Bonus score for each token in hotwords.
float hotwords_score; float hotwords_score;
float blank_penalty;
SherpaOnnxOnlineCtcFstDecoderConfig ctc_fst_decoder_config; SherpaOnnxOnlineCtcFstDecoderConfig ctc_fst_decoder_config;
const char *rule_fsts; const char *rule_fsts;