雾聪
2023-08-10 ffb05b9ae7eccc47416e9e7fae9dea54d400a245
funasr/runtime/onnxruntime/include/com-define.h
@@ -1,7 +1,6 @@
#pragma once
#ifndef COMDEFINE_H
#define COMDEFINE_H
namespace funasr {
#define S_BEGIN  0
#define S_MIDDLE 1
#define S_END    2
@@ -14,11 +13,14 @@
// 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"
@@ -43,6 +45,11 @@
#define AM_CONFIG_NAME "config.yaml"
#define PUNC_CONFIG_NAME "punc.yaml"
#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_SILENCE_DURATION 800
@@ -64,12 +71,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
@@ -77,4 +98,4 @@
#define DUN_INDEX 5
#define CACHE_POP_TRIGGER_LIMIT   200
#endif
} // namespace funasr