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)
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)