oww-rs 0.3.3

Minimalistic version of the OpenWakeWord inference in Rust and ONNX runtime
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
34
35
36
37
38
39
40
41

use crate::ModelType;
use crate::model::{Detection, Model};
use crate::oww::audio::AudioFeaturesTract;
use circular_buffer::CircularBuffer;
use std::time::Instant;

pub mod audio;
mod oww_model;

pub const OWW_MODEL_CHUNK_SIZE: usize = 1280;

const DETECTION_BUFFER_SIZE: usize = 12; // 12 detection ~ 1 secs

#[derive(Debug)]
pub struct OwwModel {
    audio: AudioFeaturesTract,
    pub tract_model: ModelType,
    threshold: f32,
    pub last_detection_time: Instant,
    // pub detections_buffer: CircularBuffer<DETECTION_BUFFER_SIZE, f32>,
    pub detections_buffer: CircularBuffer<DETECTION_BUFFER_SIZE, f32>,
    pub model_unlock_word: String,
}

impl Model for OwwModel {

    fn frame_length(&self) -> u32 {
        OWW_MODEL_CHUNK_SIZE as u32
    }

    fn detect(&mut self, chunk: Vec<f32>) -> Option<Detection> {
        // let chunk_f32 = chunk.to_vec().iter().map(|x| *x as f32).collect::<Vec<f32>>();
        Some(self.detection(chunk))
    }

    fn detect_i16(&mut self, _data: Vec<i16>) -> Option<Detection> {
        None
    }


}