face_id 0.4.1

Face detection (bounding boxes, eyes/nose/mouth markers), and facial recognition (embeddings).
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
42
43

use crate::error::FaceIdError;
#[cfg(feature = "hf-hub")]
use hf_hub::api::tokio::Api;
use std::path::PathBuf;

pub struct HfModel {
    pub id: String,
    pub file: String,
}

impl HfModel {
    #[must_use]
    pub fn default_detector() -> Self {
        Self {
            id: "RuteNL/SCRFD-face-detection-ONNX".to_owned(),
            file: "34g_gnkps.onnx".to_owned(),
        }
    }

    #[must_use]
    pub fn default_embedder() -> Self {
        Self {
            id: "public-data/insightface".to_owned(),
            file: "models/buffalo_l/w600k_r50.onnx".to_owned(),
        }
    }

    #[must_use]
    pub fn default_gender_age() -> Self {
        Self {
            id: "public-data/insightface".to_owned(),
            file: "models/buffalo_l/genderage.onnx".to_owned(),
        }
    }
}

#[cfg(feature = "hf-hub")]
pub async fn get_hf_model(model: HfModel) -> Result<PathBuf, FaceIdError> {
    let api = Api::new()?;
    let repo = api.model(model.id);

    Ok(repo.get(&model.file).await?)
}