From 74af093826b77022a8b70ec2216a96044e5be7d4 Mon Sep 17 00:00:00 2001 From: Ryan OShea Date: Fri, 7 Aug 2020 16:27:34 +0100 Subject: IVGCVSW-5159 Add Accuracy Check for YoloV3 Big App * Add Check Accuracy Method * Add Ability to pass in comparison file paths * Add compare_detection to yolo v3 class Signed-off-by: Ryan OShea Change-Id: I914ffe4805316263dc19d76a777fff6e35f44961 --- tests/TfLiteYoloV3Big-Armnn/NMS.hpp | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'tests/TfLiteYoloV3Big-Armnn/NMS.hpp') 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& 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& expected); + /** Perform Non-Maxima Supression on a list of given detections * * @param[in] config Configuration metadata for NMS -- cgit v1.2.1