insight-face-rs 0.1.0

Face detection and recognition library using ONNX Runtime (Rust)
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

use ndarray::ShapeError;
use ort::session::builder::SessionBuilder;

#[derive(Debug, thiserror::Error)]
pub enum Error {
    #[error("model load error: {0}")]
    ModelLoadError(String),
    #[error("model run error: {0}")]
    ModelRunError(String),
    #[error("align face error: {0}")]
    AlignFaceError(String),
}

impl From<ort::Error<()>> for Error {
    fn from(err: ort::Error<()>) -> Self {
        Error::ModelLoadError(err.to_string())
    }
}

impl From<ort::Error<SessionBuilder>> for Error {
    fn from(err: ort::Error<SessionBuilder>) -> Self {
        Error::ModelLoadError(err.to_string())
    }
}

impl From<ShapeError> for Error {
    fn from(err: ShapeError) -> Self {
        Error::ModelRunError(err.to_string())
    }
}

pub type Result<T, E = Error> = std::result::Result<T, E>;