aboutsummaryrefslogtreecommitdiff
path: root/tests/YoloInferenceTest.hpp
diff options
context:
space:
mode:
authortelsoa01 <telmo.soares@arm.com>2018-08-31 09:22:23 +0100
committertelsoa01 <telmo.soares@arm.com>2018-08-31 09:22:23 +0100
commitc577f2c6a3b4ddb6ba87a882723c53a248afbeba (patch)
treebd7d4c148df27f8be6649d313efb24f536b7cf34 /tests/YoloInferenceTest.hpp
parent4c7098bfeab1ffe1cdc77f6c15548d3e73274746 (diff)
downloadarmnn-c577f2c6a3b4ddb6ba87a882723c53a248afbeba.tar.gz
Release 18.08
Diffstat (limited to 'tests/YoloInferenceTest.hpp')
-rw-r--r--tests/YoloInferenceTest.hpp12
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;
}