1 files changed, 9 insertions, 0 deletions
diff --git a/tests/TfLiteYoloV3Big-Armnn/NMS.hpp b/tests/TfLiteYoloV3Big-Armnn/NMS.hpp
index fc23994f58..f5e3cf38af 100644
--- a/tests/TfLiteYoloV3Big-Armnn/NMS.hpp
+++ b/tests/TfLiteYoloV3Big-Armnn/NMS.hpp
@@ -40,6 +40,15 @@ struct Detection {
 void print_detection(std::ostream& os,
                      const std::vector<Detection>& detections);
 
+/** Compare a detection object with a vector of float values
+ *
+ * @param detection [in] Detection object
+ * @param expected  [in] Vector of expected float values
+ * @return Boolean to represent if they match or not
+ */
+bool compare_detection(const yolov3::Detection& detection,
+                       const std::vector<float>& expected);
+
 /** Perform Non-Maxima Supression on a list of given detections
  *
  * @param[in] config         Configuration metadata for NMS