diff options
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 |