Add test app to drive YoloV3 split model
Change-Id: I4877f170003d3a82f97311046c52571792ad735a
Signed-off-by: Derek Lamberti <derek.lamberti@arm.com>
diff --git a/tests/TfLiteYoloV3Big-Armnn/NMS.hpp b/tests/TfLiteYoloV3Big-Armnn/NMS.hpp
new file mode 100644
index 0000000..fc23994
--- /dev/null
+++ b/tests/TfLiteYoloV3Big-Armnn/NMS.hpp
@@ -0,0 +1,52 @@
+//
+// Copyright © 2020 Arm Ltd. All rights reserved.
+// SPDX-License-Identifier: MIT
+//
+
+#pragma once
+
+#include <ostream>
+#include <vector>
+
+namespace yolov3 {
+/** Non Maxima Suprresion configuration meta-data */
+struct NMSConfig {
+ unsigned int num_classes{0}; /**< Number of classes in the detected boxes */
+ unsigned int num_boxes{0}; /**< Number of detected boxes */
+ float confidence_threshold{0.8f}; /**< Inclusion confidence threshold for a box */
+ float iou_threshold{0.8f}; /**< Inclusion threshold for Intersection-Over-Union */
+};
+
+/** Box representation structure */
+struct Box {
+ float xmin; /**< X-pos position of the low left coordinate */
+ float xmax; /**< X-pos position of the top right coordinate */
+ float ymin; /**< Y-pos position of the low left coordinate */
+ float ymax; /**< Y-pos position of the top right coordinate */
+};
+
+/** Detection structure */
+struct Detection {
+ Box box; /**< Detection box */
+ float confidence; /**< Confidence of detection */
+ std::vector<float> classes; /**< Probability of classes */
+};
+
+/** Print identified yolo detections
+ *
+ * @param[in, out] os Output stream to print to
+ * @param[in] detections Detections to print
+ */
+void print_detection(std::ostream& os,
+ const std::vector<Detection>& detections);
+
+/** Perform Non-Maxima Supression on a list of given detections
+ *
+ * @param[in] config Configuration metadata for NMS
+ * @param[in] detected_boxes Detected boxes
+ *
+ * @return A vector with the final detections
+ */
+std::vector<Detection> nms(const NMSConfig& config,
+ const std::vector<float>& detected_boxes);
+} // namespace yolov3