diff options
author | Matthew Sloyan <matthew.sloyan@arm.com> | 2020-09-09 09:07:37 +0100 |
---|---|---|
committer | Jan Eilers <jan.eilers@arm.com> | 2020-09-17 08:31:09 +0000 |
commit | 171214c8ff275c90cd4f7fc23a34ec2c83b5ea39 (patch) | |
tree | 23fd3ee288d631c8c94bede71f89f0f1e12da862 /src/backends/reference/workloads/DetectionPostProcess.cpp | |
parent | a25886e0966a6b9433cd23595688fadb88a161b2 (diff) | |
download | armnn-171214c8ff275c90cd4f7fc23a34ec2c83b5ea39.tar.gz |
IVGCVSW-5300 Remove some boost::numeric_cast from armnn/backends
* Replaced with armnn/utility/NumericCast.hpp
* Some exclusions in reference backend
* Excluded as requires float implementation in NumericCast.hpp
Signed-off-by: Matthew Sloyan <matthew.sloyan@arm.com>
Change-Id: I9e4e9cd502c865452128fa04415fd6f250baa855
Diffstat (limited to 'src/backends/reference/workloads/DetectionPostProcess.cpp')
-rw-r--r-- | src/backends/reference/workloads/DetectionPostProcess.cpp | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/src/backends/reference/workloads/DetectionPostProcess.cpp b/src/backends/reference/workloads/DetectionPostProcess.cpp index 61a504ec6b..ce07110da9 100644 --- a/src/backends/reference/workloads/DetectionPostProcess.cpp +++ b/src/backends/reference/workloads/DetectionPostProcess.cpp @@ -6,6 +6,7 @@ #include "DetectionPostProcess.hpp" #include <armnn/utility/Assert.hpp> +#include <armnn/utility/NumericCast.hpp> #include <boost/numeric/conversion/cast.hpp> @@ -67,7 +68,7 @@ std::vector<unsigned int> NonMaxSuppression(unsigned int numBoxes, } // Sort the indices based on scores. - unsigned int numAboveThreshold = boost::numeric_cast<unsigned int>(scoresAboveThreshold.size()); + unsigned int numAboveThreshold = armnn::numeric_cast<unsigned int>(scoresAboveThreshold.size()); std::vector<unsigned int> sortedIndices = GenerateRangeK(numAboveThreshold); TopKSort(numAboveThreshold, sortedIndices.data(), scoresAboveThreshold.data(), numAboveThreshold); @@ -267,7 +268,7 @@ void DetectionPostProcess(const TensorInfo& boxEncodingsInfo, } // Select max detection numbers of the highest score across all classes - unsigned int numSelected = boost::numeric_cast<unsigned int>(selectedBoxesAfterNms.size()); + unsigned int numSelected = armnn::numeric_cast<unsigned int>(selectedBoxesAfterNms.size()); unsigned int numOutput = std::min(desc.m_MaxDetections, numSelected); // Sort the max scores among the selected indices. @@ -311,7 +312,7 @@ void DetectionPostProcess(const TensorInfo& boxEncodingsInfo, desc.m_MaxDetections, desc.m_NmsIouThreshold); - unsigned int numSelected = boost::numeric_cast<unsigned int>(selectedIndices.size()); + unsigned int numSelected = armnn::numeric_cast<unsigned int>(selectedIndices.size()); unsigned int numOutput = std::min(desc.m_MaxDetections, numSelected); AllocateOutputData(detectionBoxesInfo.GetShape()[1], numOutput, boxCorners, selectedIndices, |