285 CommandLineProcessor cmdline;
286 if (!cmdline.ProcessCommandLine(argc, argv))
290 const std::string imagePath(cmdline.GetInputFileName());
291 const std::string outputPath(cmdline.GetOutputFileName());
294 const unsigned int newWidth = cmdline.GetNewWidth();
295 const unsigned int newHeight = cmdline.GetNewHeight();
296 const unsigned int batchSize = 1;
299 std::vector<armnnUtils::TContainer> imageDataContainers;
307 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
311 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
315 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
320 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
326 ARMNN_LOG(fatal) <<
"Failed to load image file " << imagePath <<
" with error: " << e.
what();
330 std::ofstream imageTensorFile;
331 imageTensorFile.open(outputPath, std::ofstream::out);
332 if (imageTensorFile.is_open())
334 mapbox::util::apply_visitor(
336 imageDataContainers[0]
339 if (!imageTensorFile)
341 ARMNN_LOG(fatal) <<
"Failed to write to output file" << outputPath;
342 imageTensorFile.close();
345 imageTensorFile.close();
349 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)
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)
std::vector< int8_t > PrepareImageTensor< int8_t >(const std::string &imagePath, unsigned int newWidth, unsigned int newHeight, const NormalizationParameters &normParams, unsigned int batchSize, const armnn::DataLayout &outputLayout)
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)