diff options
| author | Finn Williams <Finn.Williams@arm.com> | 2019-05-16 16:40:41 +0100 |
|---|---|---|
| committer | finn.williams <finn.williams@arm.com> | 2019-05-17 13:58:41 +0000 |
| commit | b6e1756848a7e662dfdc0e97b421e2bb24db52b9 (patch) | |
| tree | 18648ca9da1254461af720cd0b5914cce34138b7 | |
| parent | 88054f855792514f033d9c3bfe0d5e4e93cac528 (diff) | |
| download | armnn-b6e1756848a7e662dfdc0e97b421e2bb24db52b9.tar.gz | |
IVGCVSW-3036 Change Armnn Quantizer to specify Armnn format
Signed-off-by: FinnWilliamsArm <Finn.Williams@arm.com>
Change-Id: I776a97ff54ea3ccc28b62afe3c512cbdc7252811
| -rw-r--r-- | src/armnnQuantizer/CommandLineProcessor.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/armnnQuantizer/CommandLineProcessor.cpp b/src/armnnQuantizer/CommandLineProcessor.cpp index 758726e72a..d2163c0869 100644 --- a/src/armnnQuantizer/CommandLineProcessor.cpp +++ b/src/armnnQuantizer/CommandLineProcessor.cpp @@ -101,7 +101,7 @@ bool CommandLineProcessor::ProcessCommandLine(int argc, char* argv[]) "Preserve the input and output data types") ("outdir,d", po::value<std::string>(&m_OutputDirectory)->required(), "Directory that output file will be written to") - ("outfile,o", po::value<std::string>(&m_OutputFileName)->required(), "Output file name"); + ("outfile,o", po::value<std::string>(&m_OutputFileName)->required(), "ArmNN output file name"); } catch (const std::exception& e) { @@ -115,8 +115,10 @@ bool CommandLineProcessor::ProcessCommandLine(int argc, char* argv[]) { po::store(po::parse_command_line(argc, argv, desc), vm); - if (vm.count("help")) + if (vm.count("help") || argc <= 1) { + std::cout << "Convert a Fp32 ArmNN model to a quantized ArmNN model." << std::endl; + std::cout << std::endl; std::cout << desc << std::endl; return false; } |