281 CommandLineProcessor cmdline;
282 if (!cmdline.ProcessCommandLine(argc, argv))
286 const std::string imagePath(cmdline.GetInputFileName());
287 const std::string outputPath(cmdline.GetOutputFileName());
290 const unsigned int newWidth = cmdline.GetNewWidth();
291 const unsigned int newHeight = cmdline.GetNewHeight();
292 const unsigned int batchSize = 1;
295 using TContainer = mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<uint8_t>>;
296 std::vector<TContainer> imageDataContainers;
304 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
308 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
313 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
319 ARMNN_LOG(fatal) <<
"Failed to load image file " << imagePath <<
" with error: " << e.
what();
323 std::ofstream imageTensorFile;
324 imageTensorFile.open(outputPath, std::ofstream::out);
325 if (imageTensorFile.is_open())
327 mapbox::util::apply_visitor(
329 imageDataContainers[0]
332 if (!imageTensorFile)
334 ARMNN_LOG(fatal) <<
"Failed to write to output file" << outputPath;
335 imageTensorFile.close();
338 imageTensorFile.close();
342 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)
mapbox::util::variant< std::vector< float >, std::vector< int >, std::vector< unsigned char >, std::vector< int8_t > > TContainer
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)