From 5c3e2192c49c5ecb3d74226d3b9a11a464cb109d Mon Sep 17 00:00:00 2001
From: lyblsgo <lyblsgo@163.com>
Date: 星期二, 25 四月 2023 14:48:31 +0800
Subject: [PATCH] add glog && tclap; rm some deps
---
funasr/runtime/onnxruntime/src/fsmn-vad.cpp | 18 ++++++++----------
1 files changed, 8 insertions(+), 10 deletions(-)
diff --git a/funasr/runtime/onnxruntime/src/fsmn-vad.cpp b/funasr/runtime/onnxruntime/src/fsmn-vad.cpp
index 0f87cb2..2988350 100644
--- a/funasr/runtime/onnxruntime/src/fsmn-vad.cpp
+++ b/funasr/runtime/onnxruntime/src/fsmn-vad.cpp
@@ -1,8 +1,6 @@
#include <fstream>
#include "precomp.h"
-//#include "glog/logging.h"
-
void FsmnVad::InitVad(const std::string &vad_model, const std::string &vad_cmvn, int vad_sample_rate, int vad_silence_duration, int vad_max_len,
float vad_speech_noise_thres) {
@@ -34,10 +32,10 @@
vad_session_ = std::make_shared<Ort::Session>(
env_, vad_model.c_str(), session_options_);
} catch (std::exception const &e) {
- //LOG(ERROR) << "Error when load onnx model: " << e.what();
+ LOG(ERROR) << "Error when load onnx model: " << e.what();
exit(0);
}
- //LOG(INFO) << "vad onnx:";
+ LOG(INFO) << "vad onnx:";
GetInputOutputInfo(vad_session_, &vad_in_names_, &vad_out_names_);
}
@@ -59,8 +57,8 @@
shape << j;
shape << " ";
}
- // LOG(INFO) << "\tInput " << i << " : name=" << name.get() << " type=" << type
- // << " dims=" << shape.str();
+ LOG(INFO) << "\tInput " << i << " : name=" << name.get() << " type=" << type
+ << " dims=" << shape.str();
(*in_names)[i] = name.get();
name.release();
}
@@ -78,8 +76,8 @@
shape << j;
shape << " ";
}
- // LOG(INFO) << "\tOutput " << i << " : name=" << name.get() << " type=" << type
- // << " dims=" << shape.str();
+ LOG(INFO) << "\tOutput " << i << " : name=" << name.get() << " type=" << type
+ << " dims=" << shape.str();
(*out_names)[i] = name.get();
name.release();
}
@@ -119,12 +117,12 @@
// 4. Onnx infer
std::vector<Ort::Value> vad_ort_outputs;
try {
- // VLOG(3) << "Start infer";
+ VLOG(3) << "Start infer";
vad_ort_outputs = vad_session_->Run(
Ort::RunOptions{nullptr}, vad_in_names_.data(), vad_inputs.data(),
vad_inputs.size(), vad_out_names_.data(), vad_out_names_.size());
} catch (std::exception const &e) {
- // LOG(ERROR) << e.what();
+ LOG(ERROR) << e.what();
return;
}
--
Gitblit v1.9.1