blob: 315ebccf075d43476b665df510302f2ce31bd942 (
plain)
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
//
// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
#pragma once
#include "BoundingBox.hpp"
#include <string>
#include <vector>
namespace od
{
/**
* An object detection network inference result decoded data representation.
*/
class DetectedObject
{
public:
DetectedObject();
/**
* Creates detection with given parameters.
*
* @param id - class id
* @param label - human readable text class label
* @param boundingBox - rectangular detection coordinates
* @param score - detection score/probability
*/
DetectedObject(unsigned int id,
std::string label,
const BoundingBox& boundingBox,
float score);
~DetectedObject() = default;
/**
* Get class id
* @return id
*/
unsigned int GetId() const;
/**
* Get human readable text class label
* @return label
*/
const std::string& GetLabel() const;
/**
* Get rectangular detection coordinates
* @return detection coordinates
*/
const BoundingBox& GetBoundingBox() const;
/**
* Get detection score
* @return score
*/
float GetScore() const;
/**
* Set class id
* @param[in] id - class id
*/
void SetId(unsigned int id);
/**
* Set class label
* @param[in] label - human readable text class label
*/
void SetLabel(const std::string& label);
/**
* Set detection coordinates
* @param[in] boundingBox detection coordinates
*/
void SetBoundingBox(const BoundingBox& boundingBox);
/**
* Set detection score
* @param[in] score - detection score
*/
void SetScore(float score);
private:
unsigned int m_Id;
std::string m_Label;
BoundingBox m_BoundingBox;
float m_Score;
};
using DetectedObjects = std::vector<DetectedObject>;
}// namespace od
|
