7 #include "../InferenceTestImage.hpp" 12 #include <boost/program_options.hpp> 13 #include <boost/variant.hpp> 27 class CommandLineProcessor
30 bool ValidateInputFile(
const std::string& inputFileName)
32 if (inputFileName.empty())
34 std::cerr <<
"No input file name specified" << std::endl;
38 if (!fs::exists(inputFileName))
40 std::cerr <<
"Input file [" << inputFileName <<
"] does not exist" << std::endl;
44 if (fs::is_directory(inputFileName))
46 std::cerr <<
"Input file [" << inputFileName <<
"] is a directory" << std::endl;
53 bool ValidateLayout(
const std::string& layout)
57 std::cerr <<
"No layout specified" << std::endl;
61 std::vector<std::string> supportedLayouts = {
"NHWC",
"NCHW" };
63 auto iterator = std::find(supportedLayouts.begin(), supportedLayouts.end(), layout);
64 if (iterator == supportedLayouts.end())
66 std::cerr <<
"Layout [" << layout <<
"] is not supported" << std::endl;
73 bool ValidateOutputFile(std::string& outputFileName)
75 if (outputFileName.empty())
77 std::cerr <<
"No output file name specified" << std::endl;
81 if (fs::exists(outputFileName))
83 std::cerr <<
"Output file [" << outputFileName <<
"] already exists" << std::endl;
87 if (fs::is_directory(outputFileName))
89 std::cerr <<
"Output file [" << outputFileName <<
"] is a directory" << std::endl;
93 fs::path outputPath(outputFileName);
94 if (!fs::exists(outputPath.parent_path()))
96 std::cerr <<
"Output directory [" << outputPath.parent_path().c_str() <<
"] does not exist" << std::endl;
103 bool ProcessCommandLine(
int argc,
char* argv[])
105 namespace po = boost::program_options;
107 po::options_description desc(
"Options");
111 (
"help,h",
"Display help messages")
112 (
"infile,i", po::value<std::string>(&m_InputFileName)->required(),
113 "Input image file to generate tensor from")
114 (
"model-format,f", po::value<std::string>(&m_ModelFormat)->required(),
115 "Format of the intended model file that uses the images." 116 "Different formats have different image normalization styles." 117 "Accepted values (caffe, tensorflow, tflite)")
118 (
"outfile,o", po::value<std::string>(&m_OutputFileName)->required(),
119 "Output raw tensor file path")
120 (
"output-type,z", po::value<std::string>(&m_OutputType)->default_value(
"float"),
121 "The data type of the output tensors." 122 "If unset, defaults to \"float\" for all defined inputs. " 123 "Accepted values (float, int or qasymm8)")
124 (
"new-width", po::value<std::string>(&m_NewWidth)->default_value(
"0"),
125 "Resize image to new width. Keep original width if unspecified")
126 (
"new-height", po::value<std::string>(&m_NewHeight)->default_value(
"0"),
127 "Resize image to new height. Keep original height if unspecified")
128 (
"layout,l", po::value<std::string>(&m_Layout)->default_value(
"NHWC"),
129 "Output data layout, \"NHWC\" or \"NCHW\", default value NHWC");
131 catch (
const std::exception& e)
133 std::cerr <<
"Fatal internal error: [" << e.what() <<
"]" << std::endl;
137 po::variables_map vm;
141 po::store(po::parse_command_line(argc, argv, desc), vm);
143 if (vm.count(
"help"))
145 std::cout << desc << std::endl;
151 catch (
const po::error& e)
153 std::cerr << e.what() << std::endl << std::endl;
154 std::cerr << desc << std::endl;
158 if (!ValidateInputFile(m_InputFileName))
163 if (!ValidateLayout(m_Layout))
168 if (!ValidateOutputFile(m_OutputFileName))
176 std::string GetInputFileName() {
return m_InputFileName;}
179 if (m_Layout ==
"NHWC")
183 else if (m_Layout ==
"NCHW")
192 std::string GetOutputFileName() {
return m_OutputFileName;}
193 unsigned int GetNewWidth() {
return static_cast<unsigned int>(std::stoi(m_NewWidth));}
194 unsigned int GetNewHeight() {
return static_cast<unsigned int>(std::stoi(m_NewHeight));}
197 if (m_ModelFormat ==
"caffe")
201 else if (m_ModelFormat ==
"tensorflow")
205 else if (m_ModelFormat ==
"tflite")
216 if (m_OutputType ==
"float")
220 else if (m_OutputType ==
"int")
224 else if (m_OutputType ==
"qasymm8")
235 std::string m_InputFileName;
236 std::string m_Layout;
237 std::string m_OutputFileName;
238 std::string m_NewWidth;
239 std::string m_NewHeight;
240 std::string m_ModelFormat;
241 std::string m_OutputType;
246 int main(
int argc,
char* argv[])
248 CommandLineProcessor cmdline;
249 if (!cmdline.ProcessCommandLine(argc, argv))
253 const std::string imagePath(cmdline.GetInputFileName());
254 const std::string outputPath(cmdline.GetOutputFileName());
257 const unsigned int newWidth = cmdline.GetNewWidth();
258 const unsigned int newHeight = cmdline.GetNewHeight();
259 const unsigned int batchSize = 1;
262 using TContainer = boost::variant<std::vector<float>, std::vector<int>, std::vector<uint8_t>>;
263 std::vector<TContainer> imageDataContainers;
271 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
275 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
280 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
286 ARMNN_LOG(fatal) <<
"Failed to load image file " << imagePath <<
" with error: " << e.
what();
290 std::ofstream imageTensorFile;
291 imageTensorFile.open(outputPath, std::ofstream::out);
292 if (imageTensorFile.is_open())
294 boost::apply_visitor([&imageTensorFile](
auto&& imageData) {
WriteImageTensorImpl(imageData, imageTensorFile); },
295 imageDataContainers[0]);
296 if (!imageTensorFile)
298 ARMNN_LOG(fatal) <<
"Failed to write to output file" << outputPath;
299 imageTensorFile.close();
302 imageTensorFile.close();
306 ARMNN_LOG(fatal) <<
"Failed to open output file" << outputPath;
NormalizationParameters GetNormalizationParameters(const SupportedFrontend &modelFormat, const armnn::DataType &outputType)
Get normalization parameters.
virtual const char * what() const noexcept override
#define ARMNN_LOG(severity)
void WriteImageTensorImpl(const std::vector< ElemType > &imageData, std::ofstream &imageTensorFile)
Write image tensor to ofstream.
std::vector< uint8_t > PrepareImageTensor< uint8_t >(const std::string &imagePath, unsigned int newWidth, unsigned int newHeight, const NormalizationParameters &normParams, unsigned int batchSize, const armnn::DataLayout &outputLayout)
boost::variant< std::vector< float >, std::vector< int >, std::vector< unsigned char > > TContainer
std::vector< int > PrepareImageTensor< int >(const std::string &imagePath, unsigned int newWidth, unsigned int newHeight, const NormalizationParameters &normParams, unsigned int batchSize, const armnn::DataLayout &outputLayout)
Base class for all ArmNN exceptions so that users can filter to just those.
int main(int argc, char *argv[])
std::vector< float > PrepareImageTensor< float >(const std::string &imagePath, unsigned int newWidth, unsigned int newHeight, const NormalizationParameters &normParams, unsigned int batchSize, const armnn::DataLayout &outputLayout)