diff options
Diffstat (limited to 'tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp')
-rw-r--r-- | tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp b/tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp index c676cd7355..4cf16d78cc 100644 --- a/tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp +++ b/tests/TfLiteMnasNet-Armnn/TfLiteMnasNet-Armnn.cpp @@ -42,7 +42,9 @@ int main(int argc, char* argv[]) dataDir, 224, 224, - imageSet); + imageSet, + 127.5f, + {0.5f,0.5f,0.5f}); }, &inputTensorShape); } |