speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update funasr-onnx-offline

Browse Source
This commit is contained in:
雾聪 2023-06-02 23:00:08 +08:00
parent 8b6a110336
commit fae856e23d
1 changed files with 10 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
funasr/runtime/onnxruntime/bin/funasr-onnx-offline.cpp
View File

@ -88,9 +88,12 @@ int main(int argc, char** argv)
// read wav_path // read wav_path
vector<string> wav_list; vector<string> wav_list;
vector<string> wav_ids;
string default_id = "wav_default_id";
string wav_path_ = model_path.at(WAV_PATH); string wav_path_ = model_path.at(WAV_PATH);
if(is_target_file(wav_path_, "wav") || is_target_file(wav_path_, "pcm")){ if(is_target_file(wav_path_, "wav") || is_target_file(wav_path_, "pcm")){
wav_list.emplace_back(wav_path_); wav_list.emplace_back(wav_path_);
wav_ids.emplace_back(default_id);
} }
else if(is_target_file(wav_path_, "scp")){ else if(is_target_file(wav_path_, "scp")){
ifstream in(wav_path_); ifstream in(wav_path_);
@ -105,6 +108,7 @@ int main(int argc, char** argv)
string column1, column2; string column1, column2;
iss >> column1 >> column2; iss >> column1 >> column2;
wav_list.emplace_back(column2); wav_list.emplace_back(column2);
wav_ids.emplace_back(column1);
} }
in.close(); in.close();
}else{ }else{
@ -114,7 +118,9 @@ int main(int argc, char** argv)
float snippet_time = 0.0f; float snippet_time = 0.0f;
long taking_micros = 0; long taking_micros = 0;
for(auto& wav_file : wav_list){ for (int i = 0; i < wav_list.size(); i++) {
auto& wav_file = wav_list[i];
auto& wav_id = wav_ids[i];
gettimeofday(&start, NULL); gettimeofday(&start, NULL);
FUNASR_RESULT result=FunOfflineInfer(asr_hanlde, wav_file.c_str(), RASR_NONE, NULL, 16000); FUNASR_RESULT result=FunOfflineInfer(asr_hanlde, wav_file.c_str(), RASR_NONE, NULL, 16000);
gettimeofday(&end, NULL); gettimeofday(&end, NULL);
@ -124,7 +130,7 @@ int main(int argc, char** argv)
if (result) if (result)
{ {
string msg = FunASRGetResult(result, 0); string msg = FunASRGetResult(result, 0);
LOG(INFO)<<"Result: "<<msg; LOG(INFO)<< wav_id <<" : "<<msg;
snippet_time += FunASRGetRetSnippetTime(result); snippet_time += FunASRGetRetSnippetTime(result);
FunASRFreeResult(result); FunASRFreeResult(result);
} }