From e5a65fb76e1f5267625579782240fc68efe6edd7 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期四, 16 十一月 2023 16:34:40 +0800
Subject: [PATCH] Merge branch 'main' of github.com:alibaba-damo-academy/FunASR add

---
 runtime/onnxruntime/src/paraformer.h |    5 ++---
 1 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/runtime/onnxruntime/src/paraformer.h b/runtime/onnxruntime/src/paraformer.h
index 58de345..b5bc46d 100644
--- a/runtime/onnxruntime/src/paraformer.h
+++ b/runtime/onnxruntime/src/paraformer.h
@@ -28,8 +28,7 @@
         void LoadConfigFromYaml(const char* filename);
         void LoadOnlineConfigFromYaml(const char* filename);
         void LoadCmvn(const char *filename);
-        vector<float> ApplyLfr(const vector<float> &in);
-        void ApplyCmvn(vector<float> *v);
+        void LfrCmvn(std::vector<std::vector<float>> &asr_feats);
 
         std::shared_ptr<Ort::Session> hw_m_session = nullptr;
         Ort::Env hw_env_;
@@ -51,7 +50,7 @@
         void InitSegDict(const std::string &seg_dict_model);
         std::vector<std::vector<float>> CompileHotwordEmbedding(std::string &hotwords);
         void Reset();
-        vector<float> FbankKaldi(float sample_rate, const float* waves, int len);
+        void FbankKaldi(float sample_rate, const float* waves, int len, std::vector<std::vector<float>> &asr_feats);
         string Forward(float* din, int len, bool input_finished=true, const std::vector<std::vector<float>> &hw_emb={{0.0}}, void* wfst_decoder=nullptr);
         string GreedySearch( float* in, int n_len, int64_t token_nums,
                              bool is_stamp=false, std::vector<float> us_alphas={0}, std::vector<float> us_cif_peak={0});

--
Gitblit v1.9.1