diff options
Diffstat (limited to 'tests/YoloDatabase.cpp')
-rw-r--r-- | tests/YoloDatabase.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/tests/YoloDatabase.cpp b/tests/YoloDatabase.cpp index 3b3e5a90d4..8b2c630205 100644 --- a/tests/YoloDatabase.cpp +++ b/tests/YoloDatabase.cpp @@ -7,13 +7,14 @@ #include <armnn/Exceptions.hpp> #include <armnn/Logging.hpp> +#include <armnn/utility/NumericCast.hpp> + #include <array> #include <cstdint> #include <tuple> #include <utility> #include <boost/format.hpp> -#include <boost/numeric/conversion/cast.hpp> #include "InferenceTestImage.hpp" @@ -73,7 +74,7 @@ YoloDatabase::YoloDatabase(const std::string& imageDir) std::unique_ptr<YoloDatabase::TTestCaseData> YoloDatabase::GetTestCaseData(unsigned int testCaseId) { - testCaseId = testCaseId % boost::numeric_cast<unsigned int>(g_PerTestCaseInputOutput.size()); + testCaseId = testCaseId % armnn::numeric_cast<unsigned int>(g_PerTestCaseInputOutput.size()); const auto& testCaseInputOutput = g_PerTestCaseInputOutput[testCaseId]; const std::string imagePath = m_ImageDir + testCaseInputOutput.first; |