Merge pull request #332 from snakers4/adamnsandle

fx https://github.com/snakers4/silero-vad/pull/329 bug
This commit is contained in:
Dimitrii Voronin
2023-04-28 14:54:44 +03:00
committed by GitHub
2 changed files with 4 additions and 5 deletions

Binary file not shown.

View File

@@ -13,11 +13,11 @@ class OnnxWrapper():
import numpy as np
global np
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():
self.session = onnxruntime.InferenceSession(path, providers=['CPUExecutionProvider'], sess_options=opts)
else:
@@ -34,7 +34,7 @@ class OnnxWrapper():
if sr != 16000 and (sr % 16000 == 0):
step = sr // 16000
x = x[::step]
x = x[:, ::step]
sr = 16000
if sr not in self.sample_rates:
@@ -291,7 +291,7 @@ def get_speech_timestamps(audio: torch.Tensor,
triggered = True
current_speech['start'] = window_size_samples * i
continue
if triggered and (window_size_samples * i) - current_speech['start'] > max_speech_samples:
if prev_end:
current_speech['end'] = prev_end
@@ -309,7 +309,6 @@ def get_speech_timestamps(audio: torch.Tensor,
prev_end = next_start = temp_end = 0
triggered = False
continue
if (speech_prob < neg_threshold) and triggered:
if not temp_end: