diff options
author | Les Bell <les.bell@arm.com> | 2019-03-28 12:26:46 +0000 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2019-03-28 13:11:14 +0000 |
commit | 10e6be471a925307dc6cae1d80377cd73878c2f2 (patch) | |
tree | 7958f3c3ffc9b7dc651e7db6ebca99c7bad150a2 /src/armnnConverter/ArmnnConverter.cpp | |
parent | e4294fd7363fd2f5a7be25d1722f20ed935d3445 (diff) | |
download | armnn-10e6be471a925307dc6cae1d80377cd73878c2f2.tar.gz |
IVGCVSW-2885 fix ArmnnConverter help text
Change-Id: I2d4ee593f5bbfb4519724d28aa8b2862bf5e5c92
Signed-off-by: Les Bell <les.bell@arm.com>
Diffstat (limited to 'src/armnnConverter/ArmnnConverter.cpp')
-rw-r--r-- | src/armnnConverter/ArmnnConverter.cpp | 21 |
1 files changed, 11 insertions, 10 deletions
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 <boost/algorithm/string/classification.hpp> #include <boost/program_options.hpp> -#include <iostream> +#include <cstdlib> #include <fstream> +#include <iostream> 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<std::vector<std::string>>()->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<std::vector<std::string>>()->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<std::vector<std::string>>()->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; } |