From 2a66366be4c2715870e4859fd5a5db6e8a9dc00a Mon Sep 17 00:00:00 2001
From: chenmengzheAAA <123789350+chenmengzheAAA@users.noreply.github.com>
Date: 星期四, 14 九月 2023 19:00:17 +0800
Subject: [PATCH] Merge pull request #956 from alibaba-damo-academy/chenmengzheAAA-patch-4
---
funasr/runtime/onnxruntime/include/vad-model.h | 15 ++++-----------
1 files changed, 4 insertions(+), 11 deletions(-)
diff --git a/funasr/runtime/onnxruntime/include/vad-model.h b/funasr/runtime/onnxruntime/include/vad-model.h
index 646a1e9..07f1833 100644
--- a/funasr/runtime/onnxruntime/include/vad-model.h
+++ b/funasr/runtime/onnxruntime/include/vad-model.h
@@ -6,22 +6,15 @@
#include <map>
#include <vector>
+namespace funasr {
class VadModel {
public:
virtual ~VadModel(){};
virtual void InitVad(const std::string &vad_model, const std::string &vad_cmvn, const std::string &vad_config, int thread_num)=0;
- virtual std::vector<std::vector<int>> Infer(const std::vector<float> &waves)=0;
- virtual void ReadModel(const char* vad_model)=0;
- virtual void LoadConfigFromYaml(const char* filename)=0;
- virtual void FbankKaldi(float sample_rate, std::vector<std::vector<float>> &vad_feats,
- const std::vector<float> &waves)=0;
- virtual std::vector<std::vector<float>> &LfrCmvn(std::vector<std::vector<float>> &vad_feats)=0;
- virtual void Forward(
- const std::vector<std::vector<float>> &chunk_feats,
- std::vector<std::vector<float>> *out_prob)=0;
- virtual void LoadCmvn(const char *filename)=0;
- virtual void InitCache()=0;
+ virtual std::vector<std::vector<int>> Infer(std::vector<float> &waves, bool input_finished=true)=0;
};
VadModel *CreateVadModel(std::map<std::string, std::string>& model_path, int thread_num);
+VadModel *CreateVadModel(void* fsmnvad_handle);
+} // namespace funasr
#endif
--
Gitblit v1.9.1