雾聪 2 лет назад
Родитель
Сommit
fae856e23d
1 измененных файлов с 10 добавлено и 4 удалено
  1. 10 4
      funasr/runtime/onnxruntime/bin/funasr-onnx-offline.cpp

+ 10 - 4
funasr/runtime/onnxruntime/bin/funasr-onnx-offline.cpp

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