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 min_speech_samples param

Browse Source
This commit is contained in:
adamnsandle
2021-01-26 11:54:09 +00:00
parent e4b7006fee
commit d1d9e6368f
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
utils.py
View File

@@ -59,6 +59,7 @@ def get_speech_ts(wav: torch.Tensor,
num_steps: int = 8,
batch_size: int = 200,
num_samples_per_window: int = 4000,
min_speech_samples: int = 10000, #samples
run_function=validate):
num_samples = num_samples_per_window
@@ -97,7 +98,7 @@ def get_speech_ts(wav: torch.Tensor,
current_speech['start'] = step * max(0, i-num_steps)
if ((sum(buffer) / len(buffer)) < neg_trig_sum) and triggered:
current_speech['end'] = step * i
if (current_speech['end'] - current_speech['start']) > 10000:
if (current_speech['end'] - current_speech['start']) > min_speech_samples:
speeches.append(current_speech)
current_speech = {}
triggered = False