| | |
|---|
| | | #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 |
|---|
| | | |
|---|
| | | // 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" |
|---|
| | | // 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_DURATION |
|---|
| | |
|---|
| | | #define DUN_INDEX 5 |
|---|
| | | #define CACHE_POP_TRIGGER_LIMIT 200 |
|---|
| | | |
|---|
| | | #endif |
|---|
| | | } // namespace funasr |
|---|
