IVGCVSW-2029 Fix fully connected layer support in TfLite Parser and implement test for TfLite VGG16 quantized
Change-Id: I2061f62f62684b963fa0f090718f1dcffe5c93ce
diff --git a/tests/ImagePreprocessor.cpp b/tests/ImagePreprocessor.cpp
index 1f29cff..8ceedd2 100644
--- a/tests/ImagePreprocessor.cpp
+++ b/tests/ImagePreprocessor.cpp
@@ -33,10 +33,16 @@
InferenceTestImage::ResizingMethods::BilinearAndNormalized,
m_Mean, m_Stddev);
+ // duplicate data across the batch
+ for (unsigned int i = 1; i < m_BatchSize; i++)
+ {
+ result.insert( result.end(), result.begin(), result.begin() + GetNumImageElements() );
+ }
+
if (m_DataFormat == DataFormat::NCHW)
{
const armnn::PermutationVector NHWCToArmNN = { 0, 2, 3, 1 };
- armnn::TensorShape dstShape({1, 3, m_Height, m_Width});
+ armnn::TensorShape dstShape({m_BatchSize, 3, m_Height, m_Width});
std::vector<float> tempImage(result.size());
armnnUtils::Permute<float>(dstShape, NHWCToArmNN, result.data(), tempImage.data());
result.swap(tempImage);