deepgeek/silero-vad
2
0
Fork 0
mirror of https://github.com/snakers4/silero-vad.git synced 2026-02-05 18:09:22 +08:00
Code Issues Packages Projects Releases Wiki Activity

add other opsets

Browse Source
This commit is contained in:
adamnsandle
2024-11-12 08:25:06 +00:00
parent d53ba1ea11
commit 6217b08bbb
5 changed files with 24 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
src/silero_vad/data/silero_vad_op13.onnx Normal file
View File

Binary file not shown.

BIN
src/silero_vad/data/silero_vad_op14.onnx Normal file
View File

Binary file not shown.

BIN
src/silero_vad/data/silero_vad_op15.onnx Normal file
View File

Binary file not shown.

19
src/silero_vad/model.py
View File

@@ -2,10 +2,21 @@ from .utils_vad import init_jit_model, OnnxWrapper
import torch
torch.set_num_threads(1)
def load_silero_vad(onnx=False):
model_name = 'silero_vad.onnx' if onnx else 'silero_vad.jit'
def load_silero_vad(onnx=False, opset_version=16):
available_ops = [13, 14, 15, 16]
if onnx and opset_version not in available_ops:
raise Exception(f'Available ONNX opset_version: {available_ops}')
if onnx:
if opset_version == 16:
model_name = 'silero_vad.onnx'
else:
model_name = f'silero_vad_op{opset_version}.onnx'
else:
model_name = 'silero_vad.jit'
package_path = "silero_vad.data"
try:
import importlib_resources as impresources
model_file_path = str(impresources.files(package_path).joinpath(model_name))
@@ -21,5 +32,5 @@ def load_silero_vad(onnx=False):
model = OnnxWrapper(model_file_path, force_onnx_cpu=True)
else:
model = init_jit_model(model_file_path)
return model