From 59e15b00ea51fee4baeea750dc11ab1952dfab1d Mon Sep 17 00:00:00 2001 From: Nina Drozd Date: Thu, 25 Apr 2019 15:45:20 +0100 Subject: IVGCVSW-2834 Add dynamic quantization via datasets * Add QuantizationDataSet class for quantization data parsed from CSV file * Add QuantizationInput for retrieving quantization data for each layer ID * Add unit tests for command line processor and QuantizationDataSet Change-Id: Iaf0a747b5f25a59a766ac04f7158e8cb7909d179 Signed-off-by: Nina Drozd --- src/armnnQuantizer/CommandLineProcessor.cpp | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'src/armnnQuantizer/CommandLineProcessor.cpp') diff --git a/src/armnnQuantizer/CommandLineProcessor.cpp b/src/armnnQuantizer/CommandLineProcessor.cpp index 16afe289f6..4f0d989d3f 100644 --- a/src/armnnQuantizer/CommandLineProcessor.cpp +++ b/src/armnnQuantizer/CommandLineProcessor.cpp @@ -149,6 +149,9 @@ bool CommandLineProcessor::ProcessCommandLine(int argc, char* argv[]) boost::filesystem::path csvFilePath(m_CsvFileName); m_CsvFileDirectory = csvFilePath.parent_path().c_str(); } + + // If CSV file is defined, create a QuantizationDataSet for specified CSV file. + m_QuantizationDataSet = QuantizationDataSet(m_CsvFileName); } if (!armnnQuantizer::ValidateOutputDirectory(m_OutputDirectory)) @@ -158,7 +161,7 @@ bool CommandLineProcessor::ProcessCommandLine(int argc, char* argv[]) std::string output(m_OutputDirectory); output.append(m_OutputFileName); - + if (boost::filesystem::exists(output)) { std::cerr << "Output file [" << output << "] already exists" << std::endl; -- cgit v1.2.1