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

Small fix

Browse Source 
Number timestamps are now in ms.
This commit is contained in:
sontref
2020-12-31 01:22:29 +00:00
parent b42d13f869
commit b9704fbe35
2 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
utils.py
View File

@@ -124,16 +124,16 @@ def get_number_ts(wav: torch.Tensor,
for i, pred in enumerate(extended_preds):
if pred == 1:
if not triggered:
cur_timing['start'] = (i * hop_length) / sample_rate
cur_timing['start'] = int((i * hop_length) / (sample_rate / 1000))
triggered = True
elif pred == 0:
if triggered:
cur_timing['end'] = (i * hop_length) / sample_rate
cur_timing['end'] = int((i * hop_length) / (sample_rate / 1000))
timings.append(cur_timing)
cur_timing = {}
triggered = False
if cur_timing:
cur_timing['end'] = len(wav) / sample_rate
cur_timing['end'] = int(len(wav) / (sample_rate / 1000))
timings.append(cur_timing)
return timings