289 CommandLineProcessor cmdline;
290 if (!cmdline.ProcessCommandLine(argc, argv))
294 const std::string imagePath(cmdline.GetInputFileName());
295 const std::string outputPath(cmdline.GetOutputFileName());
298 const unsigned int newWidth = cmdline.GetNewWidth();
299 const unsigned int newHeight = cmdline.GetNewHeight();
300 const unsigned int batchSize = 1;
303 using TContainer = mapbox::util::variant<std::vector<float>, std::vector<int>, std::vector<uint8_t>>;
304 std::vector<TContainer> imageDataContainers;
312 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
316 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
321 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
327 ARMNN_LOG(fatal) <<
"Failed to load image file " << imagePath <<
" with error: " << e.
what();
331 std::ofstream imageTensorFile;
332 imageTensorFile.open(outputPath, std::ofstream::out);
333 if (imageTensorFile.is_open())
335 mapbox::util::apply_visitor(
337 imageDataContainers[0]
340 if (!imageTensorFile)
342 ARMNN_LOG(fatal) <<
"Failed to write to output file" << outputPath;
343 imageTensorFile.close();
346 imageTensorFile.close();
350 ARMNN_LOG(fatal) <<
"Failed to open output file" << outputPath;
NormalizationParameters GetNormalizationParameters(const SupportedFrontend &modelFormat, const armnn::DataType &outputType)
Get normalization parameters.
mapbox::util::variant< std::vector< float >, std::vector< int >, std::vector< unsigned char > > TContainer
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< float > PrepareImageTensor< float >(const std::string &imagePath, unsigned int newWidth, unsigned int newHeight, const NormalizationParameters &normParams, unsigned int batchSize, const armnn::DataLayout &outputLayout)