polyvoice 0.5.2

Speaker diarization library for Rust — online and offline, ONNX-powered, ecosystem-agnostic
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

//! Offline diarization example.
//!
//! Run with:
//!     cargo run --example offline

use polyvoice::{DiarizationConfig, DummyExtractor, OfflineDiarizer};

fn main() {
    let sample_rate = 16000;
    let seconds = 10;
    let samples = vec![0.0f32; sample_rate * seconds];

    let config = DiarizationConfig::default();
    let diarizer = OfflineDiarizer::new(config);
    let extractor = DummyExtractor::new(256);

    let result = diarizer
        .run(&samples, &extractor)
        .expect("diarization failed");

    println!("Detected {} speaker(s):", result.num_speakers);
    for turn in &result.turns {
        println!(
            "  {}: {:.2}s - {:.2}s",
            turn.speaker, turn.time.start, turn.time.end
        );
    }
}