From c2e4e3c2e9be855277d9f4fa9cd0544892ff829a Mon Sep 17 00:00:00 2001 From: 游雁 <zhifu.gzf@alibaba-inc.com> Date: 星期三, 30 八月 2023 09:57:30 +0800 Subject: [PATCH] Merge branch 'main' of github.com:alibaba-damo-academy/FunASR add --- funasr/runtime/onnxruntime/include/com-define.h | 27 +++++++++++++++++++++++++++ 1 files changed, 27 insertions(+), 0 deletions(-) diff --git a/funasr/runtime/onnxruntime/include/com-define.h b/funasr/runtime/onnxruntime/include/com-define.h index 7a6345b..e08d50f 100644 --- a/funasr/runtime/onnxruntime/include/com-define.h +++ b/funasr/runtime/onnxruntime/include/com-define.h @@ -13,17 +13,21 @@ // parser option #define MODEL_DIR "model-dir" +#define OFFLINE_MODEL_DIR "model-dir" +#define ONLINE_MODEL_DIR "online-model-dir" #define VAD_DIR "vad-dir" #define PUNC_DIR "punc-dir" #define QUANTIZE "quantize" #define VAD_QUANT "vad-quant" #define PUNC_QUANT "punc-quant" +#define ASR_MODE "mode" #define WAV_PATH "wav-path" #define WAV_SCP "wav-scp" #define TXT_PATH "txt-path" #define THREAD_NUM "thread-num" #define PORT_ID "port-id" +#define HOTWORD_SEP " " // #define VAD_MODEL_PATH "vad-model" // #define VAD_CMVN_PATH "vad-cmvn" @@ -35,12 +39,21 @@ // #define PUNC_CONFIG_PATH "punc-config" #define MODEL_NAME "model.onnx" +// hotword embedding compile model +#define MODEL_EB_NAME "model_eb.onnx" #define QUANT_MODEL_NAME "model_quant.onnx" #define VAD_CMVN_NAME "vad.mvn" #define VAD_CONFIG_NAME "vad.yaml" #define AM_CMVN_NAME "am.mvn" #define AM_CONFIG_NAME "config.yaml" #define PUNC_CONFIG_NAME "punc.yaml" +#define MODEL_SEG_DICT "seg_dict" +#define HOTWORD "hotword" + +#define ENCODER_NAME "model.onnx" +#define QUANT_ENCODER_NAME "model_quant.onnx" +#define DECODER_NAME "decoder.onnx" +#define QUANT_DECODER_NAME "decoder_quant.onnx" // vad #ifndef VAD_SILENCE_DURATION @@ -63,12 +76,26 @@ #define VAD_LFR_N 1 #endif +// asr +#ifndef PARA_LFR_M +#define PARA_LFR_M 7 +#endif + +#ifndef PARA_LFR_N +#define PARA_LFR_N 6 +#endif + +#ifndef ONLINE_STEP +#define ONLINE_STEP 9600 +#endif + // punc #define UNK_CHAR "<unk>" #define TOKEN_LEN 20 #define CANDIDATE_NUM 6 #define UNKNOW_INDEX 0 +#define NOTPUNC "_" #define NOTPUNC_INDEX 1 #define COMMA_INDEX 2 #define PERIOD_INDEX 3 -- Gitblit v1.9.1