diff options
author | nikraj01 <nikhil.raj@arm.com> | 2019-04-15 09:55:15 +0100 |
---|---|---|
committer | Nikhil Raj Arm <nikhil.raj@arm.com> | 2019-04-15 09:25:25 +0000 |
commit | b1390fcea347e8e0476a961c128f2334c4d73217 (patch) | |
tree | c0b3d762548a0149e3181c07b23ad2e2dd9f06ce /tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp | |
parent | 2b03d64da2a3a39d069a7a2366f14439afb1ad39 (diff) | |
download | armnn-b1390fcea347e8e0476a961c128f2334c4d73217.tar.gz |
IVGCVSW-2959 Update TfLiteResNetV2-50-Quantized-Armnn test
Change-Id: I6cbf2a6668894b7964f98b70216f56db4c99b718
Signed-off-by: nikraj01 <nikhil.raj@arm.com>
Diffstat (limited to 'tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp')
-rw-r--r-- | tests/TfLiteResNetV2-50-Quantized-Armnn/TfLiteResNetV2-50-Quantized-Armnn.cpp | 8 |
1 files changed, 4 insertions, 4 deletions
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<DataType>; @@ -32,7 +32,7 @@ int main(int argc, char* argv[]) retVal = armnn::test::ClassifierInferenceTestMain<DatabaseType, ParserType>( 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()); |