From 2b03d64da2a3a39d069a7a2366f14439afb1ad39 Mon Sep 17 00:00:00 2001 From: Sadik Armagan Date: Fri, 12 Apr 2019 15:17:02 +0100 Subject: IVGCVSW-2955 Update the Quantizer Tool to take an additional parameter for the user to specify a CSV file Change-Id: Id56e09f147cca5c1301ec1b6bac656cd50bfd583 Signed-off-by: Sadik Armagan --- src/armnnQuantizer/CommandLineProcessor.hpp | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'src/armnnQuantizer/CommandLineProcessor.hpp') diff --git a/src/armnnQuantizer/CommandLineProcessor.hpp b/src/armnnQuantizer/CommandLineProcessor.hpp index f55e7a213f..852fcd4070 100644 --- a/src/armnnQuantizer/CommandLineProcessor.hpp +++ b/src/armnnQuantizer/CommandLineProcessor.hpp @@ -12,6 +12,7 @@ namespace armnnQuantizer // parses the command line to extract // * the input file -f containing the serialized fp32 ArmNN input graph (must exist...and be a input graph file) +// * the csv file -c detailing the paths for RAW input tensors to use for refinement // * the directory -d to place the output file into (must already exist and be writable) // * the name of the file -o the quantized ArmNN input graph will be written to (must not already exist) // * LATER: the min and max overrides to be applied to the inputs @@ -25,10 +26,14 @@ public: bool ProcessCommandLine(int argc, char* argv[]); std::string GetInputFileName() {return m_InputFileName;} + std::string GetCsvFileName() {return m_CsvFileName;} + std::string GetCsvFileDirectory() {return m_CsvFileDirectory;} std::string GetOutputDirectoryName() {return m_OutputDirectory;} std::string GetOutputFileName() {return m_OutputFileName;} private: std::string m_InputFileName; + std::string m_CsvFileName; + std::string m_CsvFileDirectory; std::string m_OutputDirectory; std::string m_OutputFileName; }; -- cgit v1.2.1