IVGCVSW-3182 Corrected expected model predictions

 * Changed expected predictions for TfLiteVGG16Quantized-Armnn.cpp

 * Changed expected predictions for TfLiteMobileNetQuantizedSoftmax-Armnn

Change-Id: I3fe0bd3f6e44655b14b10887bc40fb0df31f81c8
Signed-off-by: FinnWilliamsArm <Finn.Williams@arm.com>

2 files changed, 8 insertions, 6 deletions

diff --git a/tests/TfLiteMobileNetQuantizedSoftmax-Armnn/TfLiteMobileNetQuantizedSoftmax-Armnn.cpp b/tests/TfLiteMobileNetQuantizedSoftmax-Armnn/TfLiteMobileNetQuantizedSoftmax-Armnn.cpp
index 70828498df..f193a9870a 100644
--- a/tests/TfLiteMobileNetQuantizedSoftmax-Armnn/TfLiteMobileNetQuantizedSoftmax-Armnn.cpp
+++ b/tests/TfLiteMobileNetQuantizedSoftmax-Armnn/TfLiteMobileNetQuantizedSoftmax-Armnn.cpp
@@ -14,11 +14,13 @@ int main(int argc, char* argv[])
     try
     {
         // Coverity fix: The following code may throw an exception of type std::length_error.
+        // The model we are using incorrectly classifies the images
+        // But can still be used for benchmarking the layers.
         std::vector<ImageSet> imageSet =
                 {
-                        {"Dog.jpg", 795},
+                        {"Dog.jpg", 789},
                         {"Cat.jpg", 592},
-                        {"shark.jpg", 436},
+                        {"shark.jpg", 755},
                 };
 
         armnn::TensorShape inputTensorShape({ 1, 128, 128, 3 });
diff --git a/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp b/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp
index 2084d2d656..030f01cce4 100644
--- a/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp
+++ b/tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp
@@ -16,11 +16,11 @@ int main(int argc, char* argv[])
         // Coverity fix: The following code may throw an exception of type std::length_error.
         std::vector<ImageSet> imageSet =
         {
-            // The model we are using incorrectly classifies everything as class 699
+            // The model we are using incorrectly classifies the images
             // But can still be used for benchmarking the layers.
-            {"Dog.jpg", 669},
-            {"Cat.jpg", 669},
-            {"shark.jpg", 669},
+            {"Dog.jpg", 178},
+            {"Cat.jpg", 39},
+            {"shark.jpg", 598},
         };
 
         armnn::TensorShape inputTensorShape({ 1, 224, 224, 3 });