From b78d47f1efb3d0662fce1b8d45a9eb11b3caef02 Mon Sep 17 00:00:00 2001
From: Lizerui9926 <110582652+Lizerui9926@users.noreply.github.com>
Date: 星期三, 26 四月 2023 17:17:52 +0800
Subject: [PATCH] Merge pull request #427 from alibaba-damo-academy/dev_gflags

---
 funasr/runtime/onnxruntime/include/com-define.h |   40 +++++++++++++++++++++++++++-------------
 1 files changed, 27 insertions(+), 13 deletions(-)

diff --git a/funasr/runtime/onnxruntime/include/com-define.h b/funasr/runtime/onnxruntime/include/com-define.h
index e2c22f4..8c88517 100644
--- a/funasr/runtime/onnxruntime/include/com-define.h
+++ b/funasr/runtime/onnxruntime/include/com-define.h
@@ -12,37 +12,51 @@
 #define MODEL_SAMPLE_RATE 16000
 #endif
 
+// model path
+#define VAD_MODEL_PATH "vad-model"
+#define VAD_CMVN_PATH "vad-cmvn"
+#define VAD_CONFIG_PATH "vad-config"
+#define AM_MODEL_PATH "am-model"
+#define AM_CMVN_PATH "am-cmvn"
+#define AM_CONFIG_PATH "am-config"
+#define PUNC_MODEL_PATH "punc-model"
+#define PUNC_CONFIG_PATH "punc-config"
+#define WAV_PATH "wav-path"
+#define WAV_SCP "wav-scp"
+#define THREAD_NUM "thread-num"
+
 // vad
-#ifndef VAD_SILENCE_DYRATION
-#define VAD_SILENCE_DYRATION 15000
+#ifndef VAD_SILENCE_DURATION
+#define VAD_SILENCE_DURATION 800
 #endif
 
 #ifndef VAD_MAX_LEN
-#define VAD_MAX_LEN 800
+#define VAD_MAX_LEN 15000
 #endif
 
 #ifndef VAD_SPEECH_NOISE_THRES
 #define VAD_SPEECH_NOISE_THRES 0.9
 #endif
 
+#ifndef VAD_LFR_M
+#define VAD_LFR_M 5
+#endif
+
+#ifndef VAD_LFR_N
+#define VAD_LFR_N 1
+#endif
+
 // punc
-#define PUNC_MODEL_FILE  "punc_model.onnx"
-#define PUNC_YAML_FILE "punc.yaml"
 #define UNK_CHAR "<unk>"
+#define TOKEN_LEN     20
 
-#define  INPUT_NUM  2
-#define  INPUT_NAME1 "input"
-#define  INPUT_NAME2 "text_lengths"
-#define  OUTPUT_NAME "logits"
-#define  TOKEN_LEN     20
-
-#define  CANDIDATE_NUM   6
+#define CANDIDATE_NUM   6
 #define UNKNOW_INDEX 0
 #define NOTPUNC_INDEX 1
 #define COMMA_INDEX 2
 #define PERIOD_INDEX 3
 #define QUESTION_INDEX 4
 #define DUN_INDEX 5
-#define  CACHE_POP_TRIGGER_LIMIT   200
+#define CACHE_POP_TRIGGER_LIMIT   200
 
 #endif

--
Gitblit v1.9.1