From b1390fcea347e8e0476a961c128f2334c4d73217 Mon Sep 17 00:00:00 2001 From: nikraj01 Date: Mon, 15 Apr 2019 09:55:15 +0100 Subject: IVGCVSW-2959 Update TfLiteResNetV2-50-Quantized-Armnn test Change-Id: I6cbf2a6668894b7964f98b70216f56db4c99b718 Signed-off-by: nikraj01 --- .../TfLiteResNetV2-50-Quantized-Armnn.cpp | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'tests') diff --git a/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp b/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp index a4d0a0123f..03510f715d 100644 --- a/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp +++ b/tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp @@ -21,7 +21,7 @@ int main(int argc, char* argv[]) }; - armnn::TensorShape inputTensorShape({ 1, 299, 299, 3 }); + armnn::TensorShape inputTensorShape({ 1, 224, 224, 3 }); using DataType = uint8_t; using DatabaseType = ImagePreprocessor; @@ -32,7 +32,7 @@ int main(int argc, char* argv[]) retVal = armnn::test::ClassifierInferenceTestMain( argc, argv, - "quant_resnet_v2_50_model.tflite", // model name + "resnet_v2_50_default_minmax.tflite", // model name true, // model is binary "input", // input tensor name "output", // output tensor name @@ -43,8 +43,8 @@ int main(int argc, char* argv[]) auto inputBinding = model.GetInputBindingInfo(); return DatabaseType( dataDir, - 299, - 299, + 224, + 224, imageSet, inputBinding.second.GetQuantizationScale(), inputBinding.second.GetQuantizationOffset()); -- cgit v1.2.1