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

Merge pull request #68 from snakers4/adamnsandle

Browse Source 
fx issue https://github.com/snakers4/silero-vad/issues/67
This commit is contained in:
Alexander Veysov
2021-05-14 13:40:57 +03:00
committed by GitHub
parent 91648f32a8 b78f45a947
commit 24e0d408ab
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
utils_vad.py
View File

@@ -203,7 +203,7 @@ def get_speech_ts_adaptive(wav: torch.Tensor,
speeches: list
list containing ends and beginnings of speech chunks (in samples)
"""
num_samples = num_samples_per_window
num_steps = int(num_samples / step)
assert min_silence_samples >= step
@@ -268,9 +268,18 @@ def get_speech_ts_adaptive(wav: torch.Tensor,
if visualize_probs:
pd.DataFrame({'probs': smoothed_probs}).plot(figsize=(16, 8))
for ts in speeches:
ts['start'] = max(0, ts['start'] - speech_pad_samples)
ts['end'] += speech_pad_samples
for i, ts in enumerate(speeches):
if i == 0:
ts['start'] = max(0, ts['start'] - speech_pad_samples)
if i != len(speeches) - 1:
silence_duration = speeches[i+1]['start'] - ts['end']
if silence_duration < 2 * speech_pad_samples:
ts['end'] += silence_duration // 2
speeches[i+1]['start'] = max(0, speeches[i+1]['start'] - silence_duration // 2)
else:
ts['end'] += speech_pad_samples
else:
ts['end'] = min(len(wav), ts['end'] + speech_pad_samples)
return speeches