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/FsmnVad.h | 20 ++++++++++----------
1 files changed, 10 insertions(+), 10 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/FsmnVad.h b/funasr/runtime/onnxruntime/src/FsmnVad.h
index 69a1b98..8744604 100644
--- a/funasr/runtime/onnxruntime/src/FsmnVad.h
+++ b/funasr/runtime/onnxruntime/src/FsmnVad.h
@@ -2,24 +2,21 @@
#ifndef VAD_SERVER_FSMNVAD_H
#define VAD_SERVER_FSMNVAD_H
-#include "e2e_vad.h"
-#include "onnxruntime_cxx_api.h"
-#include "kaldi-native-fbank/csrc/feature-fbank.h"
-#include "kaldi-native-fbank/csrc/online-feature.h"
-
+#include "precomp.h"
class FsmnVad {
public:
FsmnVad();
- void test();
- void init_vad(const std::string &vad_model, const std::string &vad_cmvn, int vad_sample_rate, int vad_silence_duration, int vad_max_len,
+ void Test();
+ void InitVad(const std::string &vad_model, const std::string &vad_cmvn, int vad_sample_rate, int vad_silence_duration, int vad_max_len,
float vad_speech_noise_thres);
- std::vector<std::vector<int>> infer(const std::vector<float> &waves);
+ std::vector<std::vector<int>> Infer(const std::vector<float> &waves);
+ void Reset();
private:
- void read_model(const std::string &vad_model);
+ void ReadModel(const std::string &vad_model);
static void GetInputOutputInfo(
const std::shared_ptr<Ort::Session> &session,
@@ -34,13 +31,16 @@
const std::vector<std::vector<float>> &chunk_feats,
std::vector<std::vector<float>> *out_prob);
- void load_cmvn(const char *filename);
+ void LoadCmvn(const char *filename);
+ void InitCache();
std::shared_ptr<Ort::Session> vad_session_ = nullptr;
Ort::Env env_;
Ort::SessionOptions session_options_;
std::vector<const char *> vad_in_names_;
std::vector<const char *> vad_out_names_;
+ std::vector<std::vector<float>> in_cache_;
+
knf::FbankOptions fbank_opts;
std::vector<float> means_list;
std::vector<float> vars_list;
--
Gitblit v1.9.1