diff options
Diffstat (limited to 'tests/YoloInferenceTest.hpp')
| -rw-r--r-- | tests/YoloInferenceTest.hpp | 21 |
1 files changed, 10 insertions, 11 deletions
diff --git a/tests/YoloInferenceTest.hpp b/tests/YoloInferenceTest.hpp index 81ba0f5127..4f391a84a5 100644 --- a/tests/YoloInferenceTest.hpp +++ b/tests/YoloInferenceTest.hpp @@ -10,13 +10,13 @@ #include <armnn/utility/Assert.hpp> #include <armnn/utility/IgnoreUnused.hpp> +#include <boost/multi_array.hpp> +#include <boost/test/tools/floating_point_comparison.hpp> + #include <algorithm> #include <array> #include <utility> -#include <boost/multi_array.hpp> -#include <boost/test/tools/floating_point_comparison.hpp> - constexpr size_t YoloOutputSize = 1470; template <typename Model> @@ -187,18 +187,17 @@ public: { } - virtual void AddCommandLineOptions(boost::program_options::options_description& options) override + virtual void AddCommandLineOptions(cxxopts::Options& options, std::vector<std::string>& required) override { - namespace po = boost::program_options; - - options.add_options() - ("data-dir,d", po::value<std::string>(&m_DataDir)->required(), - "Path to directory containing test data"); + options + .allow_unrecognised_options() + .add_options() + ("d,data-dir", "Path to directory containing test data", cxxopts::value<std::string>(m_DataDir)); - Model::AddCommandLineOptions(options, m_ModelCommandLineOptions); + Model::AddCommandLineOptions(options, m_ModelCommandLineOptions, required); } - virtual bool ProcessCommandLineOptions(const InferenceTestOptions &commonOptions) override + virtual bool ProcessCommandLineOptions(const InferenceTestOptions& commonOptions) override { if (!ValidateDirectory(m_DataDir)) { |