| | |
|---|
| | | #pragma once |
|---|
| | | |
|---|
| | | #ifndef COMDEFINE_H |
|---|
| | | #define COMDEFINE_H |
|---|
| | | |
|---|
| | | namespace funasr { |
|---|
| | | #define S_BEGIN 0 |
|---|
| | | #define S_MIDDLE 1 |
|---|
| | | #define S_END 2 |
|---|
| | |
|---|
| | | |
|---|
| | | // 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" |
|---|
| | |
|---|
| | | // #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 |
|---|
| | |
|---|
| | | #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 |
|---|
| | |
|---|
| | | #define DUN_INDEX 5 |
|---|
| | | #define CACHE_POP_TRIGGER_LIMIT 200 |
|---|
| | | |
|---|
| | | #endif |
|---|
| | | } // namespace funasr |
|---|
