From 561bdbdfc0f8fd6234c5130fdc8631bf8e294bd8 Mon Sep 17 00:00:00 2001
From: 王梦迪 <73778524+di-osc@users.noreply.github.com>
Date: 星期四, 22 五月 2025 11:27:01 +0800
Subject: [PATCH] 通过缓存seg_dict，加快seaco_paraformer推理 (#2520)
---
runtime/python/onnxruntime/funasr_onnx/vad_bin.py | 7 +++----
1 files changed, 3 insertions(+), 4 deletions(-)
diff --git a/runtime/python/onnxruntime/funasr_onnx/vad_bin.py b/runtime/python/onnxruntime/funasr_onnx/vad_bin.py
index 3f21004..af4663a 100644
--- a/runtime/python/onnxruntime/funasr_onnx/vad_bin.py
+++ b/runtime/python/onnxruntime/funasr_onnx/vad_bin.py
@@ -69,7 +69,7 @@
model_file, device_id, intra_op_num_threads=intra_op_num_threads
)
self.batch_size = batch_size
- self.vad_scorer = E2EVadModel(config["model_conf"])
+ self.vad_scorer_config = config["model_conf"]
self.max_end_sil = (
max_end_sil if max_end_sil is not None else config["model_conf"]["max_end_silence_time"]
)
@@ -90,10 +90,9 @@
waveform_list = self.load_data(audio_in, self.frontend.opts.frame_opts.samp_freq)
waveform_nums = len(waveform_list)
is_final = kwargs.get("kwargs", False)
-
segments = [[]] * self.batch_size
for beg_idx in range(0, waveform_nums, self.batch_size):
-
+ vad_scorer = E2EVadModel(self.vad_scorer_config)
end_idx = min(waveform_nums, beg_idx + self.batch_size)
waveform = waveform_list[beg_idx:end_idx]
feats, feats_len = self.extract_feat(waveform)
@@ -122,7 +121,7 @@
inputs.extend(in_cache)
scores, out_caches = self.infer(inputs)
in_cache = out_caches
- segments_part = self.vad_scorer(
+ segments_part = vad_scorer(
scores,
waveform_package,
is_final=is_final,
--
Gitblit v1.9.1