From 86b22cff94c297ebd299560d8925889e1a1ed395 Mon Sep 17 00:00:00 2001 From: narugo1992 Date: Sun, 1 Sep 2024 13:33:40 +0800 Subject: [PATCH] dev(narugo): extract vad tools --- zoo/silero/vad/source.py | 15 +++------------ 1 file changed, 3 insertions(+), 12 deletions(-) diff --git a/zoo/silero/vad/source.py b/zoo/silero/vad/source.py index 8540394..95c3f00 100644 --- a/zoo/silero/vad/source.py +++ b/zoo/silero/vad/source.py @@ -3,27 +3,18 @@ from typing import Callable, Optional import numpy as np -import onnxruntime from huggingface_hub import hf_hub_download from tqdm import tqdm from soundutils.data import Sound +from soundutils.utils import open_onnx_model languages = ['ru', 'en', 'de', 'es'] class OnnxWrapper: - def __init__(self, path, force_onnx_cpu=False): - - opts = onnxruntime.SessionOptions() - opts.inter_op_num_threads = 1 - opts.intra_op_num_threads = 1 - - if force_onnx_cpu and 'CPUExecutionProvider' in onnxruntime.get_available_providers(): - self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider'], sess_options=opts) - else: - self.session = onnxruntime.InferenceSession(path, sess_options=opts) - + def __init__(self, path): + self.session = open_onnx_model(path) self._state = np.zeros((2, 1, 128), dtype=np.float32) self._context = np.zeros(0, dtype=np.float32) self._last_sr = 0