pico-detect 0.7.0

Pixel Intensity Comparison-based Object (PICO) detection library.
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

mod detection;
mod detector;
mod padding;

pub mod clusterize;
pub mod multiscale;

use derive_builder::Builder;
use image::Luma;
use pixelutil_image::ExtendedImageView;

use crate::geometry::Target;

use clusterize::Clusterizer;
use multiscale::Multiscaler;

pub use detection::Detection;
pub use detector::Detector;
pub use padding::Padding;

/// Utility for running multiscale detection with clustering and padding
/// using [`Detector`].
#[derive(Debug, Clone, Copy, Builder)]
#[builder]
pub struct DetectMultiscale {
    /// Multiscale detection parameters.
    pub multiscaler: Multiscaler,
    /// Clustering parameters.
    #[builder(default)]
    pub clusterizer: Clusterizer,
    /// Padding parameters.
    #[builder(default)]
    pub padding: Padding,
}

impl DetectMultiscale {
    /// Create default builder struct.
    #[inline]
    pub fn builder() -> DetectMultiscaleBuilder {
        Default::default()
    }

    /// Run multiscale detection with clustering and padding on the specified image.
    #[inline]
    pub fn run<I>(&self, detector: &Detector, image: &I) -> Vec<Detection<Target>>
    where
        I: ExtendedImageView<Pixel = Luma<u8>>,
    {
        let mut detections = Vec::new();

        self.multiscaler
            .run(self.padding.image_rect(image), |region| {
                if let Some(detection) = detector.detect(image, region) {
                    detections.push(detection);
                }
            });

        let mut clusters = Vec::new();

        self.clusterizer.clusterize(&mut detections, &mut clusters);

        clusters
    }
}