diff options
author | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-01-24 17:05:36 +0000 |
---|---|---|
committer | Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> | 2019-01-30 11:25:56 +0000 |
commit | 7cf0eaa26c1fb29ca9df97e4734ec7c1e10f81c4 (patch) | |
tree | 407b519ede76b235c54907fe80411970741e8a00 /tests/YoloInferenceTest.hpp | |
parent | 28d3d63cc0a33f8396b32fa8347c03912c065911 (diff) | |
download | armnn-7cf0eaa26c1fb29ca9df97e4734ec7c1e10f81c4.tar.gz |
IVGCVSW-2564 Add support for multiple input and output bindings in InferenceModel
Change-Id: I64d724367d42dca4b768b6c6e42acda714985950
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 5e2a4820fa..98a9d2f106 100644 --- a/tests/YoloInferenceTest.hpp +++ b/tests/YoloInferenceTest.hpp @@ -24,7 +24,7 @@ public: YoloTestCase(Model& model, unsigned int testCaseId, YoloTestCaseData& testCaseData) - : InferenceModelTestCase<Model>(model, testCaseId, std::move(testCaseData.m_InputImage), YoloOutputSize) + : InferenceModelTestCase<Model>(model, testCaseId, { std::move(testCaseData.m_InputImage) }, { YoloOutputSize }) , m_FloatComparer(boost::math::fpc::percent_tolerance(1.0f)) , m_TopObjectDetections(std::move(testCaseData.m_TopObjectDetections)) { @@ -34,7 +34,7 @@ public: { using Boost3dArray = boost::multi_array<float, 3>; - const std::vector<float>& output = this->GetOutput(); + const std::vector<float>& output = this->GetOutputs()[0]; BOOST_ASSERT(output.size() == YoloOutputSize); constexpr Boost3dArray::index gridSize = 7; @@ -178,7 +178,7 @@ class YoloTestCaseProvider : public IInferenceTestCaseProvider { public: template <typename TConstructModelCallable> - YoloTestCaseProvider(TConstructModelCallable constructModel) + explicit YoloTestCaseProvider(TConstructModelCallable constructModel) : m_ConstructModel(constructModel) { } |