From 0efc87352ce7d3903dbdedbfa5d01ca5e1cb19e7 Mon Sep 17 00:00:00 2001
From: Shi Xian <40013335+R1ckShi@users.noreply.github.com>
Date: 星期四, 05 十二月 2024 15:15:38 +0800
Subject: [PATCH] Merge pull request #2267 from modelscope/dev_sx2
---
runtime/onnxruntime/src/fsmn-vad.h | 7 +++----
1 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/runtime/onnxruntime/src/fsmn-vad.h b/runtime/onnxruntime/src/fsmn-vad.h
index adceb1f..dc4726a 100644
--- a/runtime/onnxruntime/src/fsmn-vad.h
+++ b/runtime/onnxruntime/src/fsmn-vad.h
@@ -28,10 +28,13 @@
std::vector<std::vector<float>> *in_cache,
bool is_final);
void Reset();
+
+ int GetVadSampleRate() { return vad_sample_rate_; };
std::shared_ptr<Ort::Session> vad_session_ = nullptr;
Ort::Env env_;
Ort::SessionOptions session_options_;
+ vector<string> m_strInputNames, m_strOutputNames;
std::vector<const char *> vad_in_names_;
std::vector<const char *> vad_out_names_;
std::vector<std::vector<float>> in_cache_;
@@ -51,10 +54,6 @@
void ReadModel(const char* vad_model);
void LoadConfigFromYaml(const char* filename);
-
- static void GetInputOutputInfo(
- const std::shared_ptr<Ort::Session> &session,
- std::vector<const char *> *in_names, std::vector<const char *> *out_names);
void FbankKaldi(float sample_rate, std::vector<std::vector<float>> &vad_feats,
std::vector<float> &waves);
--
Gitblit v1.9.1