250 CommandLineProcessor cmdline;
251 if (!cmdline.ProcessCommandLine(argc, argv))
255 const std::string imagePath(cmdline.GetInputFileName());
256 const std::string outputPath(cmdline.GetOutputFileName());
259 const unsigned int newWidth = cmdline.GetNewWidth();
260 const unsigned int newHeight = cmdline.GetNewHeight();
261 const unsigned int batchSize = 1;
264 using TContainer = boost::variant<std::vector<float>, std::vector<int>, std::vector<uint8_t>>;
265 std::vector<TContainer> imageDataContainers;
273 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
277 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
282 imagePath, newWidth, newHeight, normParams, batchSize, outputLayout));
288 ARMNN_LOG(fatal) <<
"Failed to load image file " << imagePath <<
" with error: " << e.
what();
292 std::ofstream imageTensorFile;
293 imageTensorFile.open(outputPath, std::ofstream::out);
294 if (imageTensorFile.is_open())
296 boost::apply_visitor([&imageTensorFile](
auto&& imageData) {
WriteImageTensorImpl(imageData, imageTensorFile); },
297 imageDataContainers[0]);
298 if (!imageTensorFile)
300 ARMNN_LOG(fatal) <<
"Failed to write to output file" << outputPath;
301 imageTensorFile.close();
304 imageTensorFile.close();
308 ARMNN_LOG(fatal) <<
"Failed to open output file" << outputPath;
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)
#define ARMNN_LOG(severity)
virtual const char * what() const noexcept override
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)
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)
boost::variant< std::vector< float >, std::vector< int >, std::vector< unsigned char > > TContainer
void WriteImageTensorImpl(const std::vector< ElemType > &imageData, std::ofstream &imageTensorFile)
NormalizationParameters GetNormalizationParameters(const SupportedFrontend &modelFormat, const armnn::DataType &outputType)