IVGCVSW-2437 Inference test for TensorFlow Lite MobileNet SSD

Change-Id: If7ee1efa3ee79d9eca41c5a6219b3fc42e740efe

1 files changed, 49 insertions, 0 deletions

diff --git a/tests/ObjectDetectionCommon.hpp b/tests/ObjectDetectionCommon.hpp
new file mode 100644
index 0000000000..85b54c255f
--- /dev/null
+++ b/tests/ObjectDetectionCommon.hpp
@@ -0,0 +1,49 @@
+//
+// Copyright © 2017 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+#pragma once
+
+#include <string>
+#include <utility>
+
+namespace
+{
+
+struct BoundingBox
+{
+    BoundingBox()
+        : BoundingBox(0.0f, 0.0f, 0.0f, 0.0f)
+    {}
+
+    BoundingBox(float xMin, float yMin, float xMax, float yMax)
+        : m_XMin(xMin)
+        , m_YMin(yMin)
+        , m_XMax(xMax)
+        , m_YMax(yMax)
+    {}
+
+    float m_XMin;
+    float m_YMin;
+    float m_XMax;
+    float m_YMax;
+};
+
+struct DetectedObject
+{
+    DetectedObject(unsigned int detectedClass,
+                   const BoundingBox& boundingBox,
+                   float confidence)
+        : m_Class(detectedClass)
+        , m_BoundingBox(boundingBox)
+        , m_Confidence(confidence)
+    {}
+
+    unsigned int m_Class;
+    BoundingBox  m_BoundingBox;
+    float        m_Confidence;
+};
+
+using ObjectDetectionInput = std::pair<std::string, DetectedObject>;
+
+} // anonymous namespace
\ No newline at end of file