diff options
author | surmeh01 <surabhi.mehta@arm.com> | 2018-09-05 17:09:22 +0100 |
---|---|---|
committer | Matthew Bentham <matthew.bentham@arm.com> | 2018-09-25 14:54:29 +0100 |
commit | 57f1e2076391288f964491104dfe9561733dbb3b (patch) | |
tree | 8d51a43794aeeb9d9158fb813b51494d1b1709f3 /tests/TfMobileNet-Armnn | |
parent | f195f03e095a5c4dc6880be11af64cab83b5c94b (diff) | |
download | armnn-57f1e2076391288f964491104dfe9561733dbb3b.tar.gz |
IVGCVSW-1808 : Updated the network used from Official Tf(http://download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224.tgz)
Change-Id: I3a07d7a161c8ac81c6a94cc0b3557ee7fa5e48e5
Diffstat (limited to 'tests/TfMobileNet-Armnn')
-rw-r--r-- | tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp b/tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp index 3d57666b58..44bbb70989 100644 --- a/tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp +++ b/tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp @@ -50,9 +50,9 @@ int main(int argc, char* argv[]) // Coverity fix: ClassifierInferenceTestMain() may throw uncaught exceptions. retVal = armnn::test::ClassifierInferenceTestMain<DatabaseType, ParserType>( argc, argv, - "mobilenet_v1_1.0_224_fp32.pb", // model name + "mobilenet_v1_1.0_224_frozen.pb", // model name true, // model is binary - "input", "output", // input and output tensor names + "input", "MobilenetV1/Predictions/Reshape_1", // input and output tensor names { 0, 1, 2 }, // test images to test with as above [&imageSet](const char* dataDir, const ModelType&) { // This creates a 224x224x3 NHWC float tensor to pass to Armnn |