diff options
author | Éanna Ó Catháin <eanna.ocathain@arm.com> | 2019-05-08 14:00:45 +0100 |
---|---|---|
committer | Eanna O Cathain Arm <eanna.ocathain@arm.com> | 2019-05-08 15:23:27 +0000 |
commit | a4247d5a50502811a6956dffd990c0254622b7e1 (patch) | |
tree | a2e8742695673bc8e958cce316e6ddeafcc59642 /src/armnnUtils/ModelAccuracyChecker.cpp | |
parent | c2fe5fb3a070ce2c7daebf63d0def3d57cec09d3 (diff) | |
download | armnn-a4247d5a50502811a6956dffd990c0254622b7e1.tar.gz |
IVGCVSW-2900 Adding the Accuracy Checker Tool and tests
Change-Id: I4ac325e45f2236b8e0757d21046f117024ce3979
Signed-off-by: Éanna Ó Catháin <eanna.ocathain@arm.com>
Diffstat (limited to 'src/armnnUtils/ModelAccuracyChecker.cpp')
-rw-r--r-- | src/armnnUtils/ModelAccuracyChecker.cpp | 31 |
1 files changed, 31 insertions, 0 deletions
diff --git a/src/armnnUtils/ModelAccuracyChecker.cpp b/src/armnnUtils/ModelAccuracyChecker.cpp new file mode 100644 index 0000000000..bee5ca2365 --- /dev/null +++ b/src/armnnUtils/ModelAccuracyChecker.cpp @@ -0,0 +1,31 @@ +// +// Copyright © 2017 Arm Ltd. All rights reserved. +// SPDX-License-Identifier: MIT +// + +#include <vector> +#include <map> +#include <boost/log/trivial.hpp> +#include "ModelAccuracyChecker.hpp" + +namespace armnnUtils +{ + +armnnUtils::ModelAccuracyChecker::ModelAccuracyChecker(const std::map<std::string, int>& validationLabels) + : m_GroundTruthLabelSet(validationLabels){} + +float ModelAccuracyChecker::GetAccuracy(unsigned int k) +{ + if(k > 10) { + BOOST_LOG_TRIVIAL(info) << "Accuracy Tool only supports a maximum of Top 10 Accuracy. " + "Printing Top 10 Accuracy result!"; + k = 10; + } + unsigned int total = 0; + for (unsigned int i = k; i > 0; --i) + { + total += m_TopK[i]; + } + return static_cast<float>(total * 100) / static_cast<float>(m_ImagesProcessed); +} +}
\ No newline at end of file |