commit 7cf0eaa26c1fb29ca9df97e4734ec7c1e10f81c4
author Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Thu Jan 24 17:05:36 2019 +0000
committer Aron Virginas-Tar <Aron.Virginas-Tar@arm.com> Wed Jan 30 11:25:56 2019 +0000
tree 407b519ede76b235c54907fe80411970741e8a00
parent 28d3d63cc0a33f8396b32fa8347c03912c065911

IVGCVSW-2564 Add support for multiple input and output bindings in InferenceModel

Change-Id: I64d724367d42dca4b768b6c6e42acda714985950

tests/YoloInferenceTest.hpp

diff --git a/tests/YoloInferenceTest.hpp b/tests/YoloInferenceTest.hpp
index 5e2a482..98a9d2f 100644
--- a/tests/YoloInferenceTest.hpp
+++ b/tests/YoloInferenceTest.hpp
@@ -24,7 +24,7 @@
     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 @@
     {
         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 @@
 {
 public:
     template <typename TConstructModelCallable>
-    YoloTestCaseProvider(TConstructModelCallable constructModel)
+    explicit YoloTestCaseProvider(TConstructModelCallable constructModel)
         : m_ConstructModel(constructModel)
     {
     }