diff options
author | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
---|---|---|
committer | telsoa01 <telmo.soares@arm.com> | 2018-08-31 09:22:23 +0100 |
commit | c577f2c6a3b4ddb6ba87a882723c53a248afbeba (patch) | |
tree | bd7d4c148df27f8be6649d313efb24f536b7cf34 /tests/YoloInferenceTest.hpp | |
parent | 4c7098bfeab1ffe1cdc77f6c15548d3e73274746 (diff) | |
download | armnn-c577f2c6a3b4ddb6ba87a882723c53a248afbeba.tar.gz |
Release 18.08
Diffstat (limited to 'tests/YoloInferenceTest.hpp')
-rw-r--r-- | tests/YoloInferenceTest.hpp | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/tests/YoloInferenceTest.hpp b/tests/YoloInferenceTest.hpp index edc4808939..c46cc64b73 100644 --- a/tests/YoloInferenceTest.hpp +++ b/tests/YoloInferenceTest.hpp @@ -105,10 +105,10 @@ public: { for (Boost3dArray::index c = 0; c < numClasses; ++c) { - // Resolved confidence: Class probabilities * scales + // Resolved confidence: class probabilities * scales. const float confidence = classProbabilities[y][x][c] * scales[y][x][s]; - // Resolve bounding box and store + // Resolves bounding box and stores. YoloBoundingBox box; box.m_X = boxes[y][x][s][0]; box.m_Y = boxes[y][x][s][1]; @@ -121,16 +121,16 @@ public: } } - // Sort detected objects by confidence + // Sorts detected objects by confidence. std::sort(detectedObjects.begin(), detectedObjects.end(), [](const YoloDetectedObject& a, const YoloDetectedObject& b) { - // Sort by largest confidence first, then by class + // Sorts by largest confidence first, then by class. return a.m_Confidence > b.m_Confidence || (a.m_Confidence == b.m_Confidence && a.m_Class > b.m_Class); }); - // Check the top N detections + // Checks the top N detections. auto outputIt = detectedObjects.begin(); auto outputEnd = detectedObjects.end(); @@ -138,7 +138,7 @@ public: { if (outputIt == outputEnd) { - // Somehow expected more things to check than detections found by the model + // Somehow expected more things to check than detections found by the model. return TestCaseResult::Abort; } |