diff options
author | FinnWilliamsArm <Finn.Williams@arm.com> | 2019-05-29 13:42:37 +0100 |
---|---|---|
committer | FinnWilliamsArm <Finn.Williams@arm.com> | 2019-05-30 10:22:32 +0100 |
commit | edb8b2ec42624ec27c37bc1b3d345ef0c97c024c (patch) | |
tree | 962d3fc425f7b1f7f3e1d33a7e89c66cc7ce5e58 /tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp | |
parent | d7f2908961c002c041ef6422945d840aefd0658a (diff) | |
download | armnn-edb8b2ec42624ec27c37bc1b3d345ef0c97c024c.tar.gz |
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>
Diffstat (limited to 'tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp')
-rw-r--r-- | tests/TfLiteVGG16Quantized-Armnn/TfLiteVGG16Quantized-Armnn.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
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 }); |