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

use image::GenericImageView;

///Represents generic image data. `width` and `height` are the width and height of the image and
///`pixel_buffer` is the raw pixel information of the image.
#[derive(Default, Clone)]
pub struct GenericImage {
    image_dimension: ImageDimension,
    pixel_buffer: Vec<u8>,
}

impl GenericImage {
    ///creates a new GenericImage with the given width, height, and pixel data
    pub fn new(width: u32, height: u32, pixel_buffer: Vec<u8>) -> Self {
        GenericImage {
            image_dimension: ImageDimension { width, height },
            pixel_buffer,
        }
    }
}

///Dimensions (width and height) of an image
#[derive(Clone, Copy, Default)]
pub struct ImageDimension {
    ///Width of the image.
    pub width: u32,
    ///Height of the image
    pub height: u32,
}

///Image data to be passed to the input tensor. Provides handles to the dimensions of the image and
/// the raw pixels of the image.
pub trait DetectionImage {
    ///Returns the dimensions of the image
    fn dimension(&self) -> ImageDimension;
    ///Returns a vector containing the raw pixel information.
    fn pixel_buffer(&self) -> Vec<u8>;
}

impl DetectionImage for GenericImage {
    fn dimension(&self) -> ImageDimension {
        self.image_dimension
    }

    fn pixel_buffer(&self) -> Vec<u8> {
        self.pixel_buffer.to_vec()
    }
}

impl DetectionImage for image::DynamicImage {
    fn dimension(&self) -> ImageDimension {
        let (width, height) = self.dimensions();
        ImageDimension { width, height }
    }

    fn pixel_buffer(&self) -> Vec<u8> {
        self.raw_pixels()
    }
}