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

Merge pull request #289 from Tomiinek/patch-1

Browse Source 
Fixing ONNX init
This commit is contained in:
Dimitrii Voronin
2023-02-28 15:43:18 +02:00
committed by GitHub
parent 262bcb4b40 085d76f08e
commit 4f57fae3fa
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
utils_vad.py
View File

@@ -13,12 +13,15 @@ class OnnxWrapper():
import numpy as np import numpy as np
global np global np
import onnxruntime import onnxruntime
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(): if force_onnx_cpu and 'CPUExecutionProvider' in onnxruntime.get_available_providers():
self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider']) self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider'], sess_options=opts)
else: else:
self.session = onnxruntime.InferenceSession(path) self.session = onnxruntime.InferenceSession(path, sess_options=opts)
self.session.intra_op_num_threads = 1
self.session.inter_op_num_threads = 1
self.reset_states() self.reset_states()
self.sample_rates = [8000, 16000] self.sample_rates = [8000, 16000]