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

Add CTC HLG decoding for JNI (#810)

Browse Source
This commit is contained in:
Fangjun Kuang
2024-04-25 17:20:02 +08:00
committed by GitHub
parent 6686c7d3e6
commit f7b3735621
21 changed files with 429 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
sherpa-onnx/jni/offline-recognizer.cc
View File

@@ -147,6 +147,19 @@ static OfflineRecognizerConfig GetOfflineConfig(JNIEnv *env, jobject config) {
ans.model_config.whisper.tail_paddings =
env->GetIntField(whisper_config, fid);
fid = env->GetFieldID(
model_config_cls, "nemo",
"Lcom/k2fsa/sherpa/onnx/OfflineNemoEncDecCtcModelConfig;");
jobject nemo_config = env->GetObjectField(model_config, fid);
jclass nemo_config_cls = env->GetObjectClass(nemo_config);
fid = env->GetFieldID(paraformer_config_cls, "model", "Ljava/lang/String;");
s = (jstring)env->GetObjectField(nemo_config, fid);
p = env->GetStringUTFChars(s, nullptr);
ans.model_config.nemo_ctc.model = p;
env->ReleaseStringUTFChars(s, p);
return ans;
}

16
sherpa-onnx/jni/online-recognizer.cc
View File

@@ -198,6 +198,22 @@ static OnlineRecognizerConfig GetConfig(JNIEnv *env, jobject config) {
fid = env->GetFieldID(lm_model_config_cls, "scale", "F");
ans.lm_config.scale = env->GetFloatField(lm_model_config, fid);
fid = env->GetFieldID(cls, "ctcFstDecoderConfig",
"Lcom/k2fsa/sherpa/onnx/OnlineCtcFstDecoderConfig;");
jobject fst_decoder_config = env->GetObjectField(config, fid);
jclass fst_decoder_config_cls = env->GetObjectClass(fst_decoder_config);
fid = env->GetFieldID(fst_decoder_config_cls, "graph", "Ljava/lang/String;");
s = (jstring)env->GetObjectField(fst_decoder_config, fid);
p = env->GetStringUTFChars(s, nullptr);
ans.ctc_fst_decoder_config.graph = p;
env->ReleaseStringUTFChars(s, p);
fid = env->GetFieldID(fst_decoder_config_cls, "maxActive", "I");
ans.ctc_fst_decoder_config.max_active =
env->GetIntField(fst_decoder_config, fid);
return ans;
}
} // namespace sherpa_onnx

1
sherpa-onnx/jni/wave-reader.cc
View File

@@ -6,6 +6,7 @@
#include <fstream>
#include "sherpa-onnx/csrc/macros.h"
#include "sherpa-onnx/csrc/onnx-utils.h"
#include "sherpa-onnx/jni/common.h"
static jobjectArray ReadWaveImpl(JNIEnv *env, std::istream &is,