warden-vision 0.1.2

Reusable computer vision building blocks for detection, tracking, and annotation, in the spirit of Roboflow's Python `supervision` 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

//! Draws a [`crate::geometry::PolygonZone`]'s boundary and its live
//! occupancy/in/out counts, mirroring [`crate::annotators::LineZoneAnnotator`]
//! for polygonal (rather than line) zones.

use crate::annotators::Color;

/// Draws a [`crate::geometry::PolygonZone`]'s outline and count label.
#[derive(Debug, Clone)]
pub struct PolygonZoneAnnotator {
    /// Color of the outline and its count label.
    pub color: Color,
    /// Stroke width, in pixels, of the drawn outline.
    pub thickness: u32,
    /// Relative size of the count label text.
    pub text_scale: f32,
}

impl PolygonZoneAnnotator {
    /// Creates a new polygon-zone annotator.
    pub fn new(color: Color, thickness: u32, text_scale: f32) -> Self {
        Self {
            color,
            thickness,
            text_scale,
        }
    }
}

impl Default for PolygonZoneAnnotator {
    fn default() -> Self {
        Self::new(Color::new(255, 255, 255), 2, 1.0)
    }
}