From 3d9f094e9652d4b84894c6fd4eae39a4a753b0f0 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期二, 16 五月 2023 23:48:00 +0800
Subject: [PATCH] train
---
funasr/runtime/onnxruntime/src/online-feature.h | 24 ++++++++++++++----------
1 files changed, 14 insertions(+), 10 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/online-feature.h b/funasr/runtime/onnxruntime/src/online-feature.h
index 60309ce..16e6e4b 100644
--- a/funasr/runtime/onnxruntime/src/online-feature.h
+++ b/funasr/runtime/onnxruntime/src/online-feature.h
@@ -1,33 +1,35 @@
-
+/**
+ * Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
+ * MIT License (https://opensource.org/licenses/MIT)
+ * Contributed by zhuzizyf(China Telecom).
+*/
+#pragma once
#include <vector>
#include "precomp.h"
using namespace std;
-
+namespace funasr {
class OnlineFeature {
public:
OnlineFeature(int sample_rate, knf::FbankOptions fbank_opts, int lfr_m_, int lfr_n_,
std::vector<std::vector<float>> cmvns_);
- void extractFeats(vector<vector<float>> &vad_feats, vector<float> waves, bool input_finished);
-
+ void ExtractFeats(vector<vector<float>> &vad_feats, vector<float> waves, bool input_finished);
private:
- void onlineFbank(vector<vector<float>> &vad_feats, vector<float> &waves);
-
+ void OnlineFbank(vector<vector<float>> &vad_feats, vector<float> &waves);
int OnlineLfrCmvn(vector<vector<float>> &vad_feats);
-
- static int compute_frame_num(int sample_length, int frame_sample_length, int frame_shift_sample_length) {
+
+ static int ComputeFrameNum(int sample_length, int frame_sample_length, int frame_shift_sample_length) {
int frame_num = static_cast<int>((sample_length - frame_sample_length) / frame_shift_sample_length + 1);
-
if (frame_num >= 1 && sample_length >= frame_sample_length)
return frame_num;
else
return 0;
}
- void reset_cache() {
+ void ResetCache() {
reserve_waveforms_.clear();
input_cache_.clear();
lfr_splice_cache_.clear();
@@ -52,3 +54,5 @@
bool input_finished_ = false;
};
+
+} // namespace funasr
--
Gitblit v1.9.1