ArmNN
 24.08
DetectionPostProcess.hpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 #pragma once
6 
7 #include "armnn/Tensor.hpp"
8 #include "armnn/Descriptors.hpp"
9 
10 #include "Decoders.hpp"
11 
12 #include <vector>
13 
14 namespace armnn
15 {
16 
17 void DetectionPostProcess(const TensorInfo& boxEncodingsInfo,
18  const TensorInfo& scoresInfo,
19  const TensorInfo& anchorsInfo,
20  const TensorInfo& detectionBoxesInfo,
21  const TensorInfo& detectionClassesInfo,
22  const TensorInfo& detectionScoresInfo,
23  const TensorInfo& numDetectionsInfo,
24  const DetectionPostProcessDescriptor& desc,
25  Decoder<float>& boxEncodings,
26  Decoder<float>& scores,
27  Decoder<float>& anchors,
28  float* detectionBoxes,
29  float* detectionClasses,
30  float* detectionScores,
31  float* numDetections);
32 
33 void TopKSort(unsigned int k,
34  unsigned int* indices,
35  const float* values,
36  unsigned int numElement);
37 
38 float IntersectionOverUnion(const float* boxI, const float* boxJ);
39 
40 std::vector<unsigned int> NonMaxSuppression(unsigned int numBoxes,
41  const std::vector<float>& boxCorners,
42  const std::vector<float>& scores,
43  float nmsScoreThreshold,
44  unsigned int maxDetection,
45  float nmsIouThreshold);
46 
47 } // namespace armnn
Descriptors.hpp
armnn::IntersectionOverUnion
float IntersectionOverUnion(const float *boxI, const float *boxJ)
Definition: DetectionPostProcess.cpp:29
Tensor.hpp
Decoders.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition: 01_00_quick_start.dox:6
armnn::TopKSort
void TopKSort(unsigned int k, unsigned int *indices, const float *values, unsigned int numElement)
Definition: DetectionPostProcess.cpp:23
armnn::NonMaxSuppression
std::vector< unsigned int > NonMaxSuppression(unsigned int numBoxes, const std::vector< float > &boxCorners, const std::vector< float > &scores, float nmsScoreThreshold, unsigned int maxDetection, float nmsIouThreshold)
Definition: DetectionPostProcess.cpp:48
armnn::DetectionPostProcess
void DetectionPostProcess(const TensorInfo &boxEncodingsInfo, const TensorInfo &scoresInfo, const TensorInfo &, const TensorInfo &detectionBoxesInfo, const TensorInfo &, const TensorInfo &, const TensorInfo &, const DetectionPostProcessDescriptor &desc, Decoder< float > &boxEncodings, Decoder< float > &scores, Decoder< float > &anchors, float *detectionBoxes, float *detectionClasses, float *detectionScores, float *numDetections)
Definition: DetectionPostProcess.cpp:139