MLCE-103: MDK changes for running multiple inferences qasymm8
Change-Id: I34e8e0736e133ffb5f931ce3b5f5bfa4e2c975c2
Signed-off-by: Pablo Tello <pablo.tello@arm.com>
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp
index cb6daef..72c5eda 100644
--- a/tests/InferenceModel.hpp
+++ b/tests/InferenceModel.hpp
@@ -393,6 +393,7 @@
std::vector<std::string> m_ComputeDevices;
bool m_VisualizePostOptimizationModel;
bool m_EnableFp16TurboMode;
+ std::string m_Labels;
std::vector<armnn::BackendId> GetComputeDevicesAsBackendIds()
{
@@ -417,6 +418,9 @@
("compute,c", po::value<std::vector<std::string>>(&options.m_ComputeDevices)->
default_value(defaultComputes, boost::algorithm::join(defaultComputes, ", "))->
multitoken(), backendsMessage.c_str())
+ ("labels,l", po::value<std::string>(&options.m_Labels),
+ "Text file containing one image filename - correct label pair per line, "
+ "used to test the accuracy of the network.")
("visualize-optimized-model,v",
po::value<bool>(&options.m_VisualizePostOptimizationModel)->default_value(false),
"Produce a dot file useful for visualizing the graph post optimization."