carrot_common_interface 0.1.0

The common interface for robotics and vision application
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

use image::Rgb;

pub struct Label {
    pub class_id: usize,
    pub score: f32,
}

impl Label {
    pub fn to_string(&self, label_converter: &LabelMap) -> String {
        label_converter.string_vector[self.class_id].clone()
    }

    pub fn to_color(&self, label_converter: &LabelMap) -> Rgb<u8> {
        label_converter.color_vector[self.class_id]
    }
}

pub struct LabelMap {
    string_vector: Vec<String>,
    color_vector: Vec<Rgb<u8>>,
}

impl LabelMap {
    pub fn new(string_vector: Vec<String>, color_vector: Vec<Rgb<u8>>) -> Self {
        LabelMap {
            string_vector,
            color_vector,
        }
    }

    pub fn to_id(&self, string: &str) -> usize {
        todo! {}
    }
}