From 6e69d784e4814c3dbe35e8f70c6cf4b920c8b20b Mon Sep 17 00:00:00 2001
From: 天地 <tiandiweizun@gmail.com>
Date: 星期三, 19 三月 2025 23:10:13 +0800
Subject: [PATCH] 1. bug fix:list(mean)和list(var)，由于mean和var是numpy，导致写入到文件的格式错误，参考上面的话，大概率是list(mean.tolist())，其实外层list没有必要 (#2437)
---
runtime/onnxruntime/src/fsmn-vad.cpp | 24 ++----------------------
1 files changed, 2 insertions(+), 22 deletions(-)
diff --git a/runtime/onnxruntime/src/fsmn-vad.cpp b/runtime/onnxruntime/src/fsmn-vad.cpp
index b120c19..bcef5d4 100644
--- a/runtime/onnxruntime/src/fsmn-vad.cpp
+++ b/runtime/onnxruntime/src/fsmn-vad.cpp
@@ -60,29 +60,9 @@
LOG(ERROR) << "Error when load vad onnx model: " << e.what();
exit(-1);
}
- GetInputOutputInfo(vad_session_, &vad_in_names_, &vad_out_names_);
+ GetInputNames(vad_session_.get(), m_strInputNames, vad_in_names_);
+ GetOutputNames(vad_session_.get(), m_strOutputNames, vad_out_names_);
}
-
-void FsmnVad::GetInputOutputInfo(
- const std::shared_ptr<Ort::Session> &session,
- std::vector<const char *> *in_names, std::vector<const char *> *out_names) {
- Ort::AllocatorWithDefaultOptions allocator;
- // Input info
- int num_nodes = session->GetInputCount();
- in_names->resize(num_nodes);
- for (int i = 0; i < num_nodes; ++i) {
- std::unique_ptr<char, Ort::detail::AllocatedFree> name = session->GetInputNameAllocated(i, allocator);
- (*in_names)[i] = name.get();
- }
- // Output info
- num_nodes = session->GetOutputCount();
- out_names->resize(num_nodes);
- for (int i = 0; i < num_nodes; ++i) {
- std::unique_ptr<char, Ort::detail::AllocatedFree> name = session->GetOutputNameAllocated(i, allocator);
- (*out_names)[i] = name.get();
- }
-}
-
void FsmnVad::Forward(
const std::vector<std::vector<float>> &chunk_feats,
--
Gitblit v1.9.1