dlib-face-recognition 0.3.2

Unofficial Rust wrappers to the C++ library dlib, face recognition tools
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

use std::{ops::Deref, os::raw::c_long};

#[derive(Copy, Clone, Default, Debug, PartialEq, Eq)]
#[repr(C)]
/// A 2D Point.
pub struct Point([c_long; 2]);

impl AsRef<[c_long]> for Point {
    fn as_ref(&self) -> &[c_long] {
        &self.0
    }
}

impl Deref for Point {
    type Target = [c_long; 2];

    fn deref(&self) -> &[c_long; 2] {
        &self.0
    }
}

impl Point {
    pub fn new(x: c_long, y: c_long) -> Self {
        Self([x, y])
    }

    pub fn x(&self) -> c_long {
        self.0[0]
    }

    pub fn y(&self) -> c_long {
        self.0[1]
    }
}

#[derive(Copy, Clone, Default, Debug, PartialEq)]
#[repr(C)]
/// A Rectangle.
pub struct Rectangle {
    pub left: c_long,
    pub top: c_long,
    pub right: c_long,
    pub bottom: c_long,
}

impl Rectangle {
    pub fn width(&self) -> c_long {
        self.right - self.left
    }

    pub fn height(&self) -> c_long {
        self.bottom - self.top
    }

    pub fn size(&self) -> Point {
        Point::new(self.width(), self.height())
    }

    pub fn center_x(&self) -> f64 {
        (self.left + self.right) as f64 / 2.0
    }

    pub fn center_y(&self) -> f64 {
        (self.top + self.bottom) as f64 / 2.0
    }

    pub fn center(&self) -> [f64; 2] {
        [self.center_x(), self.center_y()]
    }
}

#[test]
fn test_default_image() {
    use crate::face_detection::{FaceDetector, FaceDetectorTrait};
    use crate::matrix::ImageMatrix;

    let matrix = ImageMatrix::default();
    let face_det = FaceDetector::default();

    let locations = face_det.face_locations(&matrix);

    assert!(locations.is_empty());
    assert_eq!(locations.len(), 0);
    assert_eq!(locations.get(0), None);
}

#[test]
fn test_point() {
    let point = unsafe {
        cpp!([] -> Point as "dlib::point" {
            return dlib::point(42, -1000);
        })
    };

    assert_eq!(point, Point([42, -1000]));
}