| | |
| | | string am_model_path; |
| | | string am_cmvn_path; |
| | | string am_config_path; |
| | | string token_path; |
| | | |
| | | am_model_path = PathAppend(model_path.at(MODEL_DIR), MODEL_NAME); |
| | | if(model_path.find(QUANTIZE) != model_path.end() && model_path.at(QUANTIZE) == "true"){ |
| | |
| | | } |
| | | am_cmvn_path = PathAppend(model_path.at(MODEL_DIR), AM_CMVN_NAME); |
| | | am_config_path = PathAppend(model_path.at(MODEL_DIR), AM_CONFIG_NAME); |
| | | token_path = PathAppend(model_path.at(MODEL_DIR), TOKEN_PATH); |
| | | |
| | | Model *mm; |
| | | mm = new Paraformer(); |
| | | mm->InitAsr(am_model_path, am_cmvn_path, am_config_path, thread_num); |
| | | mm->InitAsr(am_model_path, am_cmvn_path, am_config_path, token_path, thread_num); |
| | | return mm; |
| | | }else if(type == ASR_ONLINE){ |
| | | // online |
| | |
| | | string de_model_path; |
| | | string am_cmvn_path; |
| | | string am_config_path; |
| | | string token_path; |
| | | |
| | | en_model_path = PathAppend(model_path.at(MODEL_DIR), ENCODER_NAME); |
| | | de_model_path = PathAppend(model_path.at(MODEL_DIR), DECODER_NAME); |
| | |
| | | } |
| | | am_cmvn_path = PathAppend(model_path.at(MODEL_DIR), AM_CMVN_NAME); |
| | | am_config_path = PathAppend(model_path.at(MODEL_DIR), AM_CONFIG_NAME); |
| | | token_path = PathAppend(model_path.at(MODEL_DIR), TOKEN_PATH); |
| | | |
| | | Model *mm; |
| | | mm = new Paraformer(); |
| | | mm->InitAsr(en_model_path, de_model_path, am_cmvn_path, am_config_path, thread_num); |
| | | mm->InitAsr(en_model_path, de_model_path, am_cmvn_path, am_config_path, token_path, thread_num); |
| | | return mm; |
| | | }else{ |
| | | LOG(ERROR)<<"Wrong ASR_TYPE : " << type; |