voice_activity_detector 0.2.1

Voice Activity Detector using the Silero VAD model.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

use std::mem;

use crate::{Sample, VoiceActivityDetector};

pub struct PredictState<'a, T> {
    vad: &'a mut VoiceActivityDetector,
    buffer: Vec<T>,
}

impl<'a, T> PredictState<'a, T>
where
    T: Sample,
{
    pub fn new(vad: &'a mut VoiceActivityDetector) -> Self {
        let chunk_size = vad.chunk_size();
        Self {
            vad,
            buffer: Vec::with_capacity(chunk_size),
        }
    }

    pub fn try_next(&mut self, sample: T) -> Option<(Vec<T>, f32)> {
        self.buffer.push(sample);
        if self.buffer.len() < self.vad.chunk_size() {
            return None;
        }

        let probability = self.vad.predict(self.buffer.iter().copied());
        let buffer = mem::replace(&mut self.buffer, Vec::with_capacity(self.vad.chunk_size()));

        Some((buffer, probability))
    }
}