From 57f1e2076391288f964491104dfe9561733dbb3b Mon Sep 17 00:00:00 2001 From: surmeh01 Date: Wed, 5 Sep 2018 17:09:22 +0100 Subject: 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 --- tests/TfMobileNet-Armnn/TfMobileNet-Armnn.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'tests') 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( 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 -- cgit v1.2.1