diff options
Diffstat (limited to 'tests/YoloInferenceTest.hpp')
-rw-r--r-- | tests/YoloInferenceTest.hpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/YoloInferenceTest.hpp b/tests/YoloInferenceTest.hpp index 4190e72365..6c783d3c48 100644 --- a/tests/YoloInferenceTest.hpp +++ b/tests/YoloInferenceTest.hpp @@ -7,13 +7,13 @@ #include "InferenceTest.hpp" #include "YoloDatabase.hpp" +#include <armnn/utility/Assert.hpp> #include <armnn/utility/IgnoreUnused.hpp> #include <algorithm> #include <array> #include <utility> -#include <boost/assert.hpp> #include <boost/multi_array.hpp> #include <boost/test/tools/floating_point_comparison.hpp> @@ -39,7 +39,7 @@ public: using Boost3dArray = boost::multi_array<float, 3>; const std::vector<float>& output = boost::get<std::vector<float>>(this->GetOutputs()[0]); - BOOST_ASSERT(output.size() == YoloOutputSize); + ARMNN_ASSERT(output.size() == YoloOutputSize); constexpr Boost3dArray::index gridSize = 7; constexpr Boost3dArray::index numClasses = 20; @@ -96,7 +96,7 @@ public: } } } - BOOST_ASSERT(output.data() + YoloOutputSize == outputPtr); + ARMNN_ASSERT(output.data() + YoloOutputSize == outputPtr); std::vector<YoloDetectedObject> detectedObjects; detectedObjects.reserve(gridSize * gridSize * numScales * numClasses); |