diff options
| author | Pablo Tello <pablo.tello@arm.com> | 2019-04-15 15:44:39 +0100 |
|---|---|---|
| committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-04-16 08:33:32 +0000 |
| commit | 507f39d2bc6f1537faca86633bf756831f0c770d (patch) | |
| tree | a455ac012c3cc1870eaf6e881cdd8993fe828559 /tests/InferenceModel.hpp | |
| parent | 472d4e038d15fa04701a80cd0f4413c6a9886d5d (diff) | |
| download | armnn-507f39d2bc6f1537faca86633bf756831f0c770d.tar.gz | |
MLCE-103: MDK changes for running multiple inferences qasymm8
Change-Id: I34e8e0736e133ffb5f931ce3b5f5bfa4e2c975c2
Signed-off-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'tests/InferenceModel.hpp')
| -rw-r--r-- | tests/InferenceModel.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tests/InferenceModel.hpp b/tests/InferenceModel.hpp index cb6daefa9e..72c5edac29 100644 --- a/tests/InferenceModel.hpp +++ b/tests/InferenceModel.hpp @@ -393,6 +393,7 @@ public: 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 @@ public: ("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." |