samples/ObjectDetection/include/Types.hpp


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

//
// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//

#pragma once

#include <cstddef>
#include <cstdint>
#include <vector>
#include <tuple>
#include <armnn/BackendId.hpp>

namespace od
{

struct Size
{

    uint32_t m_Width;
    uint32_t m_Height;

    Size() : Size(0, 0) {}

    Size(uint32_t width, uint32_t height) :
            m_Width{width}, m_Height{height} {}

    Size(const Size& other)
            : Size(other.m_Width, other.m_Height) {}

    ~Size() = default;

    Size &operator=(const Size& other) = default;
};

struct BBoxColor
{
    std::tuple<int, int, int> colorCode;
};

struct ODPipelineOptions
{
    std::string m_ModelName;
    std::string m_ModelFilePath;
    std::vector<armnn::BackendId> m_backends;
};

using InferenceResult = std::vector<float>;
using InferenceResults = std::vector<InferenceResult>;
}