6 #include "../InferenceTestImage.hpp" 21 std::array<float, 3>
mean{ { 0.0, 0.0, 0.0 } };
22 std::array<float, 3>
stddev{ { 1.0, 1.0, 1.0 } };
44 normParams.
scale = 1.0;
45 normParams.
mean = { 0.0, 0.0, 0.0 };
46 normParams.
stddev = { 1.0, 1.0, 1.0 };
57 normParams.
scale = 127.5;
58 normParams.
mean = { 1.0, 1.0, 1.0 };
61 normParams.
mean = { 128.0, 128.0, 128.0 };
81 template <
typename ElemType>
83 unsigned int newWidth,
84 unsigned int newHeight,
86 unsigned int batchSize = 1,
92 unsigned int newWidth,
93 unsigned int newHeight,
95 unsigned int batchSize,
99 std::vector<float> imageData;
107 newHeight = testImage.GetHeight();
111 imageData = testImage.Resize(newWidth, newHeight,
CHECK_LOCATION(),
119 std::vector<float> tempImage(imageData.size());
120 armnnUtils::Permute(dstShape, NHWCToArmNN, imageData.data(), tempImage.data(),
sizeof(float));
121 imageData.swap(tempImage);
129 unsigned int newWidth,
130 unsigned int newHeight,
132 unsigned int batchSize,
136 std::vector<float> imageDataFloat =
139 std::vector<int> imageDataInt;
140 imageDataInt.reserve(imageDataFloat.size());
141 std::transform(imageDataFloat.begin(), imageDataFloat.end(), std::back_inserter(imageDataInt),
142 [](
float val) {
return static_cast<int>(val); });
149 unsigned int newWidth,
150 unsigned int newHeight,
152 unsigned int batchSize,
156 std::vector<float> imageDataFloat =
158 std::vector<uint8_t> imageDataQasymm8;
159 imageDataQasymm8.reserve(imageDataFloat.size());
161 std::transform(imageDataFloat.begin(), imageDataFloat.end(), std::back_inserter(imageDataQasymm8),
162 [](
float val) {
return static_cast<uint8_t
>(val); });
163 return imageDataQasymm8;
171 template <
typename ElemType>
174 std::copy(imageData.begin(), imageData.end(), std::ostream_iterator<ElemType>(imageTensorFile,
" "));
182 std::copy(imageData.begin(), imageData.end(), std::ostream_iterator<int>(imageTensorFile,
" "));
void Permute(const armnn::TensorShape &dstShape, const armnn::PermutationVector &mappings, const void *src, void *dst, size_t dataTypeSize)
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::array< float, 3 > mean
std::vector< ElemType > PrepareImageTensor(const std::string &imagePath, unsigned int newWidth, unsigned int newHeight, const NormalizationParameters &normParams, unsigned int batchSize=1, const armnn::DataLayout &outputLayout=armnn::DataLayout::NHWC)
unsigned int GetWidth() const
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)
void WriteImageTensorImpl< uint8_t >(const std::vector< uint8_t > &imageData, std::ofstream &imageTensorFile)
std::array< float, 3 > stddev
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)
void WriteImageTensorImpl(const std::vector< ElemType > &imageData, std::ofstream &imageTensorFile)
NormalizationParameters GetNormalizationParameters(const SupportedFrontend &modelFormat, const armnn::DataType &outputType)
const armnn::PermutationVector NHWCToArmNN