diff options
Diffstat (limited to 'tests/MobileNetSsdDatabase.hpp')
-rw-r--r-- | tests/MobileNetSsdDatabase.hpp | 13 |
1 files changed, 6 insertions, 7 deletions
diff --git a/tests/MobileNetSsdDatabase.hpp b/tests/MobileNetSsdDatabase.hpp index 349d1ad96a..1a99ed715a 100644 --- a/tests/MobileNetSsdDatabase.hpp +++ b/tests/MobileNetSsdDatabase.hpp @@ -2,24 +2,23 @@ // Copyright © 2017 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // + #pragma once +#include "InferenceTestImage.hpp" #include "ObjectDetectionCommon.hpp" -#include <memory> -#include <string> -#include <vector> +#include <QuantizeHelper.hpp> #include <armnn/TypesUtils.hpp> -#include <backendsCommon/test/QuantizeHelper.hpp> #include <boost/log/trivial.hpp> #include <boost/numeric/conversion/cast.hpp> #include <array> +#include <memory> #include <string> - -#include "InferenceTestImage.hpp" +#include <vector> namespace { @@ -97,7 +96,7 @@ std::unique_ptr<MobileNetSsdTestCaseData> MobileNetSsdDatabase::GetTestCaseData( // Get image data as a vector of floats std::vector<float> floatImageData = GetImageDataAsNormalizedFloats(ImageChannelLayout::Rgb, image); - imageData = QuantizedVector<uint8_t>(m_Scale, m_Offset, floatImageData); + imageData = armnnUtils::QuantizedVector<uint8_t>(floatImageData, m_Scale, m_Offset); } catch (const InferenceTestImageException& e) { |