From 17aeb722956e493a3ec624087e1aa481e0c0d4ff Mon Sep 17 00:00:00 2001 From: lyblsgo <lyblsgo@163.com> Date: 星期一, 24 四月 2023 11:34:15 +0800 Subject: [PATCH] rename ct/vad files --- funasr/runtime/onnxruntime/include/ComDefine.h | 38 ++++++++++++++++++++++++++++++++++++++ 1 files changed, 38 insertions(+), 0 deletions(-) diff --git a/funasr/runtime/onnxruntime/include/ComDefine.h b/funasr/runtime/onnxruntime/include/ComDefine.h index f131e5e..72a843d 100644 --- a/funasr/runtime/onnxruntime/include/ComDefine.h +++ b/funasr/runtime/onnxruntime/include/ComDefine.h @@ -8,4 +8,42 @@ #define S_ALL 3 #define S_ERR 4 +#ifndef MODEL_SAMPLE_RATE +#define MODEL_SAMPLE_RATE 16000 +#endif + +// vad +#ifndef VAD_SILENCE_DYRATION +#define VAD_SILENCE_DYRATION 15000 +#endif + +#ifndef VAD_MAX_LEN +#define VAD_MAX_LEN 800 +#endif + +#ifndef VAD_SPEECH_NOISE_THRES +#define VAD_SPEECH_NOISE_THRES 0.9 +#endif + +// punc +#define PUNC_MODEL_FILE "punc_model.onnx" +#define PUNC_YAML_FILE "punc.yaml" + +#define UNK_CHAR "<unk>" + +#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 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 + #endif -- Gitblit v1.9.1