CommandLineProcessor Class Reference

#include <CommandLineProcessor.hpp>

Public Member Functions
bool	ProcessCommandLine (int argc, char *argv[])
std::string	GetInputFileName ()
std::string	GetCsvFileName ()
std::string	GetCsvFileDirectory ()
std::string	GetOutputDirectoryName ()
std::string	GetOutputFileName ()
std::string	GetQuantizationScheme ()
QuantizationDataSet	GetQuantizationDataSet ()
bool	HasPreservedDataType ()
bool	HasQuantizationData ()

Protected Attributes
std::string	m_InputFileName
std::string	m_CsvFileName
std::string	m_CsvFileDirectory
std::string	m_OutputDirectory
std::string	m_OutputFileName
std::string	m_QuantizationScheme
QuantizationDataSet	m_QuantizationDataSet
bool	m_PreserveDataType

Detailed Description

Definition at line 25 of file CommandLineProcessor.hpp.

Member Function Documentation

GetCsvFileDirectory()

std::string GetCsvFileDirectory ( )

inline

Definition at line 32 of file CommandLineProcessor.hpp.

References CommandLineProcessor::m_CsvFileDirectory.

{return m_CsvFileDirectory;}

GetCsvFileName()

std::string GetCsvFileName ( )

inline

Definition at line 31 of file CommandLineProcessor.hpp.

References CommandLineProcessor::m_CsvFileName.

{return m_CsvFileName;}

GetInputFileName()

std::string GetInputFileName ( )

inline

Definition at line 30 of file CommandLineProcessor.hpp.

References CommandLineProcessor::m_InputFileName.

Referenced by main().

{return m_InputFileName;}

GetOutputDirectoryName()

std::string GetOutputDirectoryName ( )

inline

Definition at line 33 of file CommandLineProcessor.hpp.

References CommandLineProcessor::m_OutputDirectory.

Referenced by main().

{return m_OutputDirectory;}

GetOutputFileName()

std::string GetOutputFileName ( )

inline

Definition at line 34 of file CommandLineProcessor.hpp.

References CommandLineProcessor::m_OutputFileName.

Referenced by main().

{return m_OutputFileName;}

GetQuantizationDataSet()

QuantizationDataSet GetQuantizationDataSet ( )

inline

Definition at line 36 of file CommandLineProcessor.hpp.

References CommandLineProcessor::m_QuantizationDataSet.

Referenced by main().

{return m_QuantizationDataSet;}

GetQuantizationScheme()

std::string GetQuantizationScheme ( )

inline

Definition at line 35 of file CommandLineProcessor.hpp.

References CommandLineProcessor::m_QuantizationScheme.

Referenced by main().

{return m_QuantizationScheme;}

HasPreservedDataType()

bool HasPreservedDataType ( )

inline

Definition at line 37 of file CommandLineProcessor.hpp.

References CommandLineProcessor::m_PreserveDataType.

Referenced by main().

{return m_PreserveDataType;}

HasQuantizationData()

bool HasQuantizationData ( )

inline

Definition at line 38 of file CommandLineProcessor.hpp.

References QuantizationDataSet::IsEmpty(), and CommandLineProcessor::m_QuantizationDataSet.

Referenced by main().

{return !m_QuantizationDataSet.IsEmpty();}

ProcessCommandLine()

bool ProcessCommandLine	(	int	argc,
		char *	argv[]
	)

Definition at line 85 of file CommandLineProcessor.cpp.

References CommandLineProcessor::m_CsvFileDirectory, CommandLineProcessor::m_CsvFileName, CommandLineProcessor::m_InputFileName, CommandLineProcessor::m_OutputDirectory, CommandLineProcessor::m_OutputFileName, CommandLineProcessor::m_PreserveDataType, CommandLineProcessor::m_QuantizationDataSet, CommandLineProcessor::m_QuantizationScheme, armnnQuantizer::ValidateOutputDirectory(), armnnQuantizer::ValidateProvidedFile(), and armnnQuantizer::ValidateQuantizationScheme().

Referenced by main().

{
    namespace po = boost::program_options;

    po::options_description desc("Options");
    try
    {
        desc.add_options()
                ("help,h", "Display help messages")
                ("infile,f", po::value<std::string>(&m_InputFileName)->required(),
                             "Input file containing float 32 ArmNN Input Graph")
                ("scheme,s", po::value<std::string>(&m_QuantizationScheme)->default_value("QAsymm8"),
                              "Quantization scheme, \"QAsymm8\" or \"QSymm16\", default value QAsymm8")
                ("csvfile,c", po::value<std::string>(&m_CsvFileName)->default_value(""),
                             "CSV file containing paths for RAW input tensors")
                ("preserve-data-type,p", po::bool_switch(&m_PreserveDataType)->default_value(false),
                              "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(), "ArmNN output file name");
    }
    catch (const std::exception& e)
    {
        std::cerr << "Fatal internal error: [" << e.what() << "]" << std::endl;
        return false;
    }

    po::variables_map vm;

    try
    {
        po::store(po::parse_command_line(argc, argv, desc), vm);

        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;
        }

        po::notify(vm);
    }
    catch (const po::error& e)
    {
        std::cerr << e.what() << std::endl << std::endl;
        std::cerr << desc << std::endl;
        return false;
    }

    if (!armnnQuantizer::ValidateProvidedFile(m_InputFileName))
    {
        return false;
    }

    if (!ValidateQuantizationScheme(m_QuantizationScheme))
    {
        return false;
    }

    if (m_CsvFileName != "")
    {
        if (!armnnQuantizer::ValidateProvidedFile(m_CsvFileName))
        {
            return false;
        }
        else
        {
            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))
    {
        return false;
    }

    std::string output(m_OutputDirectory);
    output.append(m_OutputFileName);

    if (boost::filesystem::exists(output))
    {
        std::cerr << "Output file [" << output << "] already exists" << std::endl;
        return false;
    }

    return true;
}

Member Data Documentation

m_CsvFileDirectory

std::string m_CsvFileDirectory

protected

Definition at line 43 of file CommandLineProcessor.hpp.

Referenced by CommandLineProcessor::GetCsvFileDirectory(), and CommandLineProcessor::ProcessCommandLine().

m_CsvFileName

std::string m_CsvFileName

protected

Definition at line 42 of file CommandLineProcessor.hpp.

Referenced by CommandLineProcessor::GetCsvFileName(), and CommandLineProcessor::ProcessCommandLine().

m_InputFileName

std::string m_InputFileName

protected

Definition at line 41 of file CommandLineProcessor.hpp.

Referenced by CommandLineProcessor::GetInputFileName(), and CommandLineProcessor::ProcessCommandLine().

m_OutputDirectory

std::string m_OutputDirectory

protected

Definition at line 44 of file CommandLineProcessor.hpp.

Referenced by CommandLineProcessor::GetOutputDirectoryName(), and CommandLineProcessor::ProcessCommandLine().

m_OutputFileName

std::string m_OutputFileName

protected

Definition at line 45 of file CommandLineProcessor.hpp.

Referenced by CommandLineProcessor::GetOutputFileName(), and CommandLineProcessor::ProcessCommandLine().

m_PreserveDataType

bool m_PreserveDataType

protected

Definition at line 48 of file CommandLineProcessor.hpp.

Referenced by CommandLineProcessor::HasPreservedDataType(), and CommandLineProcessor::ProcessCommandLine().

m_QuantizationDataSet

QuantizationDataSet m_QuantizationDataSet

protected

Definition at line 47 of file CommandLineProcessor.hpp.

Referenced by CommandLineProcessor::GetQuantizationDataSet(), CommandLineProcessor::HasQuantizationData(), and CommandLineProcessor::ProcessCommandLine().

m_QuantizationScheme

std::string m_QuantizationScheme

protected

Definition at line 46 of file CommandLineProcessor.hpp.

Referenced by CommandLineProcessor::GetQuantizationScheme(), and CommandLineProcessor::ProcessCommandLine().

---

The documentation for this class was generated from the following files:

• src/armnnQuantizer/CommandLineProcessor.hpp
• src/armnnQuantizer/CommandLineProcessor.cpp