20 const TensorInfo& detectionBoxesInfo,
21 const TensorInfo& detectionClassesInfo,
22 const TensorInfo& detectionScoresInfo,
23 const TensorInfo& numDetectionsInfo,
24 const DetectionPostProcessDescriptor& desc,
28 float* detectionBoxes,
29 float* detectionClasses,
30 float* detectionScores,
31 float* numDetections);
34 unsigned int* indices,
36 unsigned int numElement);
41 const std::vector<float>& boxCorners,
42 const std::vector<float>&
scores,
43 float nmsScoreThreshold,
44 unsigned int maxDetection,
45 float nmsIouThreshold);
void DetectionPostProcess(const TensorInfo &boxEncodingsInfo, const TensorInfo &scoresInfo, const TensorInfo &anchorsInfo, const TensorInfo &detectionBoxesInfo, const TensorInfo &detectionClassesInfo, const TensorInfo &detectionScoresInfo, const TensorInfo &numDetectionsInfo, const DetectionPostProcessDescriptor &desc, Decoder< float > &boxEncodings, Decoder< float > &scores, Decoder< float > &anchors, float *detectionBoxes, float *detectionClasses, float *detectionScores, float *numDetections)
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)
armnn::TensorInfo scoresInfo({ 1, 6, 3 }, armnn::DataType::Float32)
std::vector< float > scores({ 0.0f, 0.9f, 0.8f, 0.0f, 0.75f, 0.72f, 0.0f, 0.6f, 0.5f, 0.0f, 0.93f, 0.95f, 0.0f, 0.5f, 0.4f, 0.0f, 0.3f, 0.2f })
std::vector< float > boxEncodings({ 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f, 0.0f, -1.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f })
void TopKSort(unsigned int k, unsigned int *indices, const float *values, unsigned int numElement)
std::vector< float > anchors({ 0.5f, 0.5f, 1.0f, 1.0f, 0.5f, 0.5f, 1.0f, 1.0f, 0.5f, 0.5f, 1.0f, 1.0f, 0.5f, 10.5f, 1.0f, 1.0f, 0.5f, 10.5f, 1.0f, 1.0f, 0.5f, 100.5f, 1.0f, 1.0f })
armnn::TensorInfo anchorsInfo({ 6, 4 }, armnn::DataType::Float32)
float IntersectionOverUnion(const float *boxI, const float *boxJ)