From aae1e4f40de608ec325ee97a5836a077e796f0b9 Mon Sep 17 00:00:00 2001 From: Abin Thomas Date: Mon, 22 Jan 2024 11:08:24 +0530 Subject: [PATCH] fix window size offset bug --- utils_vad.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils_vad.py b/utils_vad.py index faf56bd..3d056b9 100644 --- a/utils_vad.py +++ b/utils_vad.py @@ -498,7 +498,7 @@ class VADIterator: if (speech_prob >= self.threshold) and not self.triggered: self.triggered = True - speech_start = self.current_sample - self.speech_pad_samples + speech_start = self.current_sample - self.speech_pad_samples - window_size_samples return {'start': int(speech_start) if not return_seconds else round(speech_start / self.sampling_rate, 1)} if (speech_prob < self.threshold - 0.15) and self.triggered: @@ -507,7 +507,7 @@ class VADIterator: if self.current_sample - self.temp_end < self.min_silence_samples: return None else: - speech_end = self.temp_end + self.speech_pad_samples + speech_end = self.temp_end + self.speech_pad_samples - window_size_samples self.temp_end = 0 self.triggered = False return {'end': int(speech_end) if not return_seconds else round(speech_end / self.sampling_rate, 1)}