baklava 0.1.5

A rust wrapper around the InsightFace library to perform face comparison
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
42
43
44

use baklava::InsightFace;
use std::sync::{Arc, Mutex};
use std::thread;

fn main() -> Result<(), Box<dyn std::error::Error>> {
    let handler = InsightFace::new("./Megatron", Some(4))?;
    let handler_safe = Arc::new(Mutex::new(handler));

    let images = vec!["./face1_test.png", "./face2_test.png"];

    let (tx, rx) = std::sync::mpsc::channel();
    let (txc, rxc) = std::sync::mpsc::channel();

    thread::spawn(move || {
        let harc = handler_safe.clone();

        while let Ok(image) = rx.recv() {
            let mut inner = harc.lock().unwrap();

            let (cosine, percentage) = inner
                .prepare_images(&[image])
                .expect("Failed to prepare images")
                .prepare_target_image("./face1_test.png")
                .unwrap()
                .compare_images(baklava::Methodology::Mean)
                .expect("Failed to compare images");

            txc.send((cosine, percentage))
                .expect("Failed to send results");

            println!("Cosine: {cosine}, Percentage: {percentage}");
        }
    });

    tx.send(images[0]).unwrap();
    tx.send(images[1]).unwrap();
    drop(tx); // Close the channel to signal the thread to finish

    while let Ok((cosine, percentage)) = rxc.recv() {
        println!("Received from thread: Cosine: {cosine}, Percentage: {percentage}");
    }

    Ok(())
}