xref: /aosp_15_r20/external/armnn/samples/ObjectDetection/src/DetectedObject.cpp (revision 89c4ff92f2867872bb9e2354d150bf0c8c502810)
1*89c4ff92SAndroid Build Coastguard Worker //
2*89c4ff92SAndroid Build Coastguard Worker // Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3*89c4ff92SAndroid Build Coastguard Worker // SPDX-License-Identifier: MIT
4*89c4ff92SAndroid Build Coastguard Worker //
5*89c4ff92SAndroid Build Coastguard Worker 
6*89c4ff92SAndroid Build Coastguard Worker #include "DetectedObject.hpp"
7*89c4ff92SAndroid Build Coastguard Worker 
8*89c4ff92SAndroid Build Coastguard Worker namespace od
9*89c4ff92SAndroid Build Coastguard Worker {
10*89c4ff92SAndroid Build Coastguard Worker 
DetectedObject()11*89c4ff92SAndroid Build Coastguard Worker DetectedObject::DetectedObject() :
12*89c4ff92SAndroid Build Coastguard Worker         DetectedObject(0u, "", BoundingBox(), 0u)
13*89c4ff92SAndroid Build Coastguard Worker {}
14*89c4ff92SAndroid Build Coastguard Worker 
DetectedObject(unsigned int id,std::string label,const BoundingBox & boundingBox,float score)15*89c4ff92SAndroid Build Coastguard Worker DetectedObject::DetectedObject(
16*89c4ff92SAndroid Build Coastguard Worker         unsigned int id,
17*89c4ff92SAndroid Build Coastguard Worker         std::string label,
18*89c4ff92SAndroid Build Coastguard Worker         const BoundingBox &boundingBox,
19*89c4ff92SAndroid Build Coastguard Worker         float score) :
20*89c4ff92SAndroid Build Coastguard Worker         m_Id(id),
21*89c4ff92SAndroid Build Coastguard Worker         m_Label(std::move(label)),
22*89c4ff92SAndroid Build Coastguard Worker         m_BoundingBox(boundingBox),
23*89c4ff92SAndroid Build Coastguard Worker         m_Score(score)
24*89c4ff92SAndroid Build Coastguard Worker {}
25*89c4ff92SAndroid Build Coastguard Worker 
GetId() const26*89c4ff92SAndroid Build Coastguard Worker unsigned int DetectedObject::GetId() const
27*89c4ff92SAndroid Build Coastguard Worker {
28*89c4ff92SAndroid Build Coastguard Worker     return m_Id;
29*89c4ff92SAndroid Build Coastguard Worker }
30*89c4ff92SAndroid Build Coastguard Worker 
GetLabel() const31*89c4ff92SAndroid Build Coastguard Worker const std::string &DetectedObject::GetLabel() const
32*89c4ff92SAndroid Build Coastguard Worker {
33*89c4ff92SAndroid Build Coastguard Worker     return m_Label;
34*89c4ff92SAndroid Build Coastguard Worker }
35*89c4ff92SAndroid Build Coastguard Worker 
GetBoundingBox() const36*89c4ff92SAndroid Build Coastguard Worker const BoundingBox &DetectedObject::GetBoundingBox() const
37*89c4ff92SAndroid Build Coastguard Worker {
38*89c4ff92SAndroid Build Coastguard Worker     return m_BoundingBox;
39*89c4ff92SAndroid Build Coastguard Worker }
40*89c4ff92SAndroid Build Coastguard Worker 
GetScore() const41*89c4ff92SAndroid Build Coastguard Worker float DetectedObject::GetScore() const
42*89c4ff92SAndroid Build Coastguard Worker {
43*89c4ff92SAndroid Build Coastguard Worker     return m_Score;
44*89c4ff92SAndroid Build Coastguard Worker }
45*89c4ff92SAndroid Build Coastguard Worker 
SetId(unsigned int id)46*89c4ff92SAndroid Build Coastguard Worker void DetectedObject::SetId(unsigned int id)
47*89c4ff92SAndroid Build Coastguard Worker {
48*89c4ff92SAndroid Build Coastguard Worker     m_Id = id;
49*89c4ff92SAndroid Build Coastguard Worker }
50*89c4ff92SAndroid Build Coastguard Worker 
SetLabel(const std::string & label)51*89c4ff92SAndroid Build Coastguard Worker void DetectedObject::SetLabel(const std::string &label)
52*89c4ff92SAndroid Build Coastguard Worker {
53*89c4ff92SAndroid Build Coastguard Worker     m_Label = label;
54*89c4ff92SAndroid Build Coastguard Worker }
55*89c4ff92SAndroid Build Coastguard Worker 
SetBoundingBox(const BoundingBox & boundingBox)56*89c4ff92SAndroid Build Coastguard Worker void DetectedObject::SetBoundingBox(const BoundingBox &boundingBox)
57*89c4ff92SAndroid Build Coastguard Worker {
58*89c4ff92SAndroid Build Coastguard Worker     m_BoundingBox = boundingBox;
59*89c4ff92SAndroid Build Coastguard Worker }
60*89c4ff92SAndroid Build Coastguard Worker 
SetScore(float score)61*89c4ff92SAndroid Build Coastguard Worker void DetectedObject::SetScore(float score)
62*89c4ff92SAndroid Build Coastguard Worker {
63*89c4ff92SAndroid Build Coastguard Worker     m_Score = score;
64*89c4ff92SAndroid Build Coastguard Worker }
65*89c4ff92SAndroid Build Coastguard Worker }// namespace od