From fa0356b81dd3c99a2bd573d1f60d7b7131e00533 Mon Sep 17 00:00:00 2001
From: lyblsgo <lyblsgo@163.com>
Date: 星期一, 24 四月 2023 11:23:40 +0800
Subject: [PATCH] rename src/e2e_vad.h
---
funasr/runtime/onnxruntime/src/paraformer_onnx.cpp | 9 ++++-----
1 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp b/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp
index 69d1554..2dd7439 100644
--- a/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp
+++ b/funasr/runtime/onnxruntime/src/paraformer_onnx.cpp
@@ -3,7 +3,7 @@
using namespace std;
using namespace paraformer;
-ModelImp::ModelImp(const char* path,int nNumThread, bool quantize, bool use_vad)
+ModelImp::ModelImp(const char* path,int nNumThread, bool quantize, bool use_vad, bool use_punc)
:env_(ORT_LOGGING_LEVEL_ERROR, "paraformer"),sessionOptions{}{
string model_path;
string cmvn_path;
@@ -14,11 +14,11 @@
string vad_path = pathAppend(path, "vad_model.onnx");
string mvn_path = pathAppend(path, "vad.mvn");
vadHandle = make_unique<FsmnVad>();
- vadHandle->init_vad(vad_path, mvn_path, MODEL_SAMPLE_RATE, VAD_MAX_LEN, VAD_SILENCE_DYRATION, VAD_SPEECH_NOISE_THRES);
+ vadHandle->InitVad(vad_path, mvn_path, MODEL_SAMPLE_RATE, VAD_MAX_LEN, VAD_SILENCE_DYRATION, VAD_SPEECH_NOISE_THRES);
}
// PUNC model
- if(true){
+ if(use_punc){
puncHandle = make_unique<CTTransformer>(path, nNumThread);
}
@@ -55,7 +55,6 @@
m_session = std::make_unique<Ort::Session>(env_, model_path.c_str(), sessionOptions);
#endif
- vector<string> m_strInputNames, m_strOutputNames;
string strName;
getInputName(m_session.get(), strName);
m_strInputNames.push_back(strName.c_str());
@@ -86,7 +85,7 @@
}
vector<std::vector<int>> ModelImp::vad_seg(std::vector<float>& pcm_data){
- return vadHandle->infer(pcm_data);
+ return vadHandle->Infer(pcm_data);
}
string ModelImp::AddPunc(const char* szInput){
--
Gitblit v1.9.1