From 7ebfaac337c3cb43052f4759aa6bfd4eec596e04 Mon Sep 17 00:00:00 2001 From: 游雁 <zhifu.gzf@alibaba-inc.com> Date: 星期四, 27 四月 2023 21:52:09 +0800 Subject: [PATCH] docs --- funasr/runtime/onnxruntime/include/com-define.h | 42 ++++++++++++++++++++++++++++-------------- 1 files changed, 28 insertions(+), 14 deletions(-) diff --git a/funasr/runtime/onnxruntime/include/com-define.h b/funasr/runtime/onnxruntime/include/com-define.h index 72a843d..9b7b212 100644 --- a/funasr/runtime/onnxruntime/include/com-define.h +++ b/funasr/runtime/onnxruntime/include/com-define.h @@ -12,38 +12,52 @@ #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" +#define PORT_ID "port-id" + // 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