1 //
2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "RefDetectionPostProcessWorkload.hpp"
7
8 #include "Decoders.hpp"
9 #include "DetectionPostProcess.hpp"
10 #include "Profiling.hpp"
11 #include "RefWorkloadUtils.hpp"
12
13 namespace armnn
14 {
15
RefDetectionPostProcessWorkload(const DetectionPostProcessQueueDescriptor & descriptor,const WorkloadInfo & info)16 RefDetectionPostProcessWorkload::RefDetectionPostProcessWorkload(
17 const DetectionPostProcessQueueDescriptor& descriptor, const WorkloadInfo& info)
18 : RefBaseWorkload<DetectionPostProcessQueueDescriptor>(descriptor, info),
19 m_Anchors(std::make_unique<ScopedTensorHandle>(*(descriptor.m_Anchors))) {}
20
Execute() const21 void RefDetectionPostProcessWorkload::Execute() const
22 {
23 Execute(m_Data.m_Inputs, m_Data.m_Outputs);
24 }
25
ExecuteAsync(ExecutionData & executionData)26 void RefDetectionPostProcessWorkload::ExecuteAsync(ExecutionData& executionData)
27 {
28 WorkingMemDescriptor* workingMemDescriptor = static_cast<WorkingMemDescriptor*>(executionData.m_Data);
29 Execute(workingMemDescriptor->m_Inputs, workingMemDescriptor->m_Outputs);
30 }
31
Execute(std::vector<ITensorHandle * > inputs,std::vector<ITensorHandle * > outputs) const32 void RefDetectionPostProcessWorkload::Execute(std::vector<ITensorHandle*> inputs,
33 std::vector<ITensorHandle*> outputs) const
34 {
35 ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefDetectionPostProcessWorkload_Execute");
36
37 const TensorInfo& boxEncodingsInfo = GetTensorInfo(inputs[0]);
38 const TensorInfo& scoresInfo = GetTensorInfo(inputs[1]);
39 const TensorInfo& anchorsInfo = m_Anchors->GetTensorInfo();
40
41 const TensorInfo& detectionBoxesInfo = GetTensorInfo(outputs[0]);
42 const TensorInfo& detectionClassesInfo = GetTensorInfo(outputs[1]);
43 const TensorInfo& detectionScoresInfo = GetTensorInfo(outputs[2]);
44 const TensorInfo& numDetectionsInfo = GetTensorInfo(outputs[3]);
45
46 auto boxEncodings = MakeDecoder<float>(boxEncodingsInfo, inputs[0]->Map());
47 auto scores = MakeDecoder<float>(scoresInfo, inputs[1]->Map());
48 auto anchors = MakeDecoder<float>(anchorsInfo, m_Anchors->Map(false));
49
50 float* detectionBoxes = reinterpret_cast<float*>(outputs[0]->Map());
51 float* detectionClasses = reinterpret_cast<float*>(outputs[1]->Map());
52 float* detectionScores = reinterpret_cast<float*>(outputs[2]->Map());
53 float* numDetections = reinterpret_cast<float*>(outputs[3]->Map());
54
55 DetectionPostProcess(boxEncodingsInfo, scoresInfo, anchorsInfo,
56 detectionBoxesInfo, detectionClassesInfo,
57 detectionScoresInfo, numDetectionsInfo, m_Data.m_Parameters,
58 *boxEncodings, *scores, *anchors, detectionBoxes,
59 detectionClasses, detectionScores, numDetections);
60 }
61
62 } //namespace armnn
63