diff options
author | Ryan OShea <Ryan.OShea2@arm.com> | 2020-08-07 16:27:34 +0100 |
---|---|---|
committer | Jim Flynn <jim.flynn@arm.com> | 2020-08-18 10:51:48 +0000 |
commit | 74af093826b77022a8b70ec2216a96044e5be7d4 (patch) | |
tree | e3e4d2bdc6e155bc6c04f1efb7848e2c1b17ab80 /tests/TfLiteYoloV3Big-Armnn/NMS.hpp | |
parent | c83eb25255c46ccce97e705e18210f0eaf36339d (diff) | |
download | armnn-74af093826b77022a8b70ec2216a96044e5be7d4.tar.gz |
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 <Ryan.OShea2@arm.com>
Change-Id: I914ffe4805316263dc19d76a777fff6e35f44961
Diffstat (limited to 'tests/TfLiteYoloV3Big-Armnn/NMS.hpp')
-rw-r--r-- | tests/TfLiteYoloV3Big-Armnn/NMS.hpp | 9 |
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 |