1 files changed, 32 insertions, 0 deletions
diff --git a/samples/ObjectDetection/include/SSDResultDecoder.hpp b/samples/ObjectDetection/include/SSDResultDecoder.hpp
new file mode 100644
index 0000000000..65afb8d376
--- /dev/null
+++ b/samples/ObjectDetection/include/SSDResultDecoder.hpp
@@ -0,0 +1,32 @@
+//
+// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+#pragma once
+
+#include "Types.hpp"
+#include "DetectedObject.hpp"
+#include "IDetectionResultDecoder.hpp"
+
+namespace od
+{
+
+class SSDResultDecoder : public IDetectionResultDecoder
+{
+public:
+    /**
+     * Constructs MobileNet ssd v1 inference results decoder.
+     *
+     * @param ObjectThreshold object score threshold
+     */
+    SSDResultDecoder(float ObjectThreshold);
+
+    DetectedObjects Decode(const InferenceResults& results,
+                           const Size& outputFrameSize,
+                           const Size& resizedFrameSize,
+                           const std::vector<std::string>& labels) override;
+
+private:
+    float m_objectThreshold;
+};
+}// namespace od
+\ No newline at end of file