From 0bb5d87d1ee98289bbe241e1f2caf1ab8e64c69c Mon Sep 17 00:00:00 2001
From: lyblsgo <lyblsgo@163.com>
Date: 星期六, 22 四月 2023 20:34:07 +0800
Subject: [PATCH] Merge branch 'dev_knf' of https://github.com/alibaba-damo-academy/FunASR into dev_knf
---
funasr/runtime/onnxruntime/src/FsmnVad.h | 6 +++++-
1 files changed, 5 insertions(+), 1 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/FsmnVad.h b/funasr/runtime/onnxruntime/src/FsmnVad.h
index 69a1b98..78302ae 100644
--- a/funasr/runtime/onnxruntime/src/FsmnVad.h
+++ b/funasr/runtime/onnxruntime/src/FsmnVad.h
@@ -1,3 +1,4 @@
+// Collaborators: zhuzizyf(China Telecom Shanghai)
#ifndef VAD_SERVER_FSMNVAD_H
#define VAD_SERVER_FSMNVAD_H
@@ -7,7 +8,6 @@
#include "kaldi-native-fbank/csrc/feature-fbank.h"
#include "kaldi-native-fbank/csrc/online-feature.h"
-
class FsmnVad {
public:
FsmnVad();
@@ -16,6 +16,7 @@
float vad_speech_noise_thres);
std::vector<std::vector<int>> infer(const std::vector<float> &waves);
+ void Reset();
private:
@@ -35,12 +36,15 @@
std::vector<std::vector<float>> *out_prob);
void load_cmvn(const char *filename);
+ void init_cache();
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