From 10e6be471a925307dc6cae1d80377cd73878c2f2 Mon Sep 17 00:00:00 2001 From: Les Bell Date: Thu, 28 Mar 2019 12:26:46 +0000 Subject: IVGCVSW-2885 fix ArmnnConverter help text Change-Id: I2d4ee593f5bbfb4519724d28aa8b2862bf5e5c92 Signed-off-by: Les Bell --- src/armnnConverter/ArmnnConverter.cpp | 21 +++++++++++---------- 1 file changed, 11 insertions(+), 10 deletions(-) (limited to 'src/armnnConverter/ArmnnConverter.cpp') diff --git a/src/armnnConverter/ArmnnConverter.cpp b/src/armnnConverter/ArmnnConverter.cpp index 9bc6cc841f..67246fcd95 100644 --- a/src/armnnConverter/ArmnnConverter.cpp +++ b/src/armnnConverter/ArmnnConverter.cpp @@ -28,8 +28,9 @@ #include #include -#include +#include #include +#include namespace { @@ -131,20 +132,20 @@ int ParseCommandLineArgs(int argc, const char* argv[], #if defined(ARMNN_ONNX_PARSER) ", onnx-binary, onnx-text" #endif -#if defined(ARMNN_TENSORFLOW_PARSER) +#if defined(ARMNN_TF_PARSER) ", tensorflow-binary, tensorflow-text" #endif #if defined(ARMNN_TF_LITE_PARSER) ", tflite-binary" #endif ".") - ("model-path,m", po::value(&modelPath)->required(), "Path to model file") + ("model-path,m", po::value(&modelPath)->required(), "Path to model file.") ("input-name,i", po::value>()->multitoken(), - "Identifier of the input tensors in the network separated by whitespace") + "Identifier of the input tensors in the network, separated by whitespace.") ("input-tensor-shape,s", po::value>()->multitoken(), - "The shape of the input tensor in the network as a flat array of integers separated by comma" - "Multiple shapes are separated by whitespace" - "This parameter is optional, depending on the network.") + "The shape of the input tensor in the network as a flat array of integers, separated by comma." + " Multiple shapes are separated by whitespace." + " This parameter is optional, depending on the network.") ("output-name,o", po::value>()->multitoken(), "Identifier of the output tensor in the network.") ("output-path,p", po::value(&outputPath)->required(), "Path to serialize the network to."); @@ -156,10 +157,10 @@ int ParseCommandLineArgs(int argc, const char* argv[], if (CheckOption(vm, "help") || argc <= 1) { - std::cout << "Convert a neural network model from provided file to ArmNN format " << std::endl; + std::cout << "Convert a neural network model from provided file to ArmNN format." << std::endl; std::cout << std::endl; std::cout << desc << std::endl; - return EXIT_SUCCESS; + exit(EXIT_SUCCESS); } po::notify(vm); } @@ -478,7 +479,7 @@ int main(int argc, const char* argv[]) } else { - BOOST_LOG_TRIVIAL(fatal) << "Unknown model format: '" << modelFormat; + BOOST_LOG_TRIVIAL(fatal) << "Unknown model format: '" << modelFormat << "'"; return EXIT_FAILURE; } -- cgit v1.2.1