From ff83116369ba6afe56430cb0b9d977fbe7f4bced Mon Sep 17 00:00:00 2001
From: Atomie CHEN <atomic_cwh@163.com>
Date: 星期一, 25 三月 2024 09:37:26 +0800
Subject: [PATCH] fix: resolve IndexError when using spk model and the audio contains only 1 segment (#1535)
---
runtime/onnxruntime/src/fsmn-vad-online.h | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/runtime/onnxruntime/src/fsmn-vad-online.h b/runtime/onnxruntime/src/fsmn-vad-online.h
index 9191304..4c82d11 100644
--- a/runtime/onnxruntime/src/fsmn-vad-online.h
+++ b/runtime/onnxruntime/src/fsmn-vad-online.h
@@ -21,6 +21,8 @@
std::vector<std::vector<int>> Infer(std::vector<float> &waves, bool input_finished);
void ExtractFeats(float sample_rate, vector<vector<float>> &vad_feats, vector<float> &waves, bool input_finished);
void Reset();
+ int GetVadSampleRate() { return vad_sample_rate_; };
+
// 2pass
std::unique_ptr<Audio> audio_handle = nullptr;
--
Gitblit v1.9.1