IVGCVSW-2437 Inference test for TensorFlow Lite MobileNet SSD
Change-Id: If7ee1efa3ee79d9eca41c5a6219b3fc42e740efe
diff --git a/tests/ObjectDetectionCommon.hpp b/tests/ObjectDetectionCommon.hpp
new file mode 100644
index 0000000..85b54c2
--- /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