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-online-vad

Browse Source
This commit is contained in:
雾聪 2023-06-02 22:34:42 +08:00
parent eee6af2ece
commit 0a0aa61ca9
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
funasr/runtime/onnxruntime/bin/funasr-onnx-online-vad.cpp
View File

@ -39,11 +39,11 @@ void GetValue(TCLAP::ValueArg<std::string>& value_arg, string key, std::map<std:
}
}
void print_segs(vector<vector<int>>* vec) {
void print_segs(vector<vector<int>>* vec, string &wav_id) {
if((*vec).size() == 0){
return;
}
string seg_out="[";
string seg_out=wav_id + ": [";
for (int i = 0; i < vec->size(); i++) {
vector<int> inner_vec = (*vec)[i];
if(inner_vec.size() == 0){
@ -104,9 +104,12 @@ int main(int argc, char *argv[])
// read wav_path
vector<string> wav_list;
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_);
@ -120,7 +123,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{
@ -131,7 +135,9 @@ int main(int argc, char *argv[])
FUNASR_HANDLE online_hanlde=FsmnVadOnlineInit(vad_hanlde);
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];
int32_t sampling_rate_ = -1;
funasr::Audio audio(1);
@ -172,7 +178,7 @@ int main(int argc, char *argv[])
if (result)
{
vector<std::vector<int>>* vad_segments = FsmnVadGetResult(result, 0);
print_segs(vad_segments);
print_segs(vad_segments, wav_id);
snippet_time += FsmnVadGetRetSnippetTime(result);
FsmnVadFreeResult(result);
}