| | |
|---|
| | | #pragma once |
|---|
| | | |
|---|
| | | #ifndef COMDEFINE_H |
|---|
| | | #define COMDEFINE_H |
|---|
| | | |
|---|
| | | namespace funasr { |
|---|
| | | #define S_BEGIN 0 |
|---|
| | | #define S_MIDDLE 1 |
|---|
| | | #define S_END 2 |
|---|
| | |
|---|
| | | #define MODEL_SAMPLE_RATE 16000 |
|---|
| | | #endif |
|---|
| | | |
|---|
| | | // parser option |
|---|
| | | #define MODEL_DIR "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 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 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 MODEL_NAME "model.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" |
|---|
| | | |
|---|
| | | // 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 |
|---|
| | | } // namespace funasr |
|---|
