diff options
author | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-02 22:04:39 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-16 14:27:27 +0100 |
commit | 6dd178f2395b34cfb360eabb0130c19ed258f5fa (patch) | |
tree | 6814e4729bbd1e652d8b9c18c9db28f3bc2f8a8a /tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp | |
parent | ed7fce413410d15c501ea52f9e6bfbbf71b3daf1 (diff) | |
download | armnn-6dd178f2395b34cfb360eabb0130c19ed258f5fa.tar.gz |
IVGCVSW-5720 Remove the Caffe Parser from ArmNN
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: Ib00be204f549efa9aa5971ecf65c2dec4a10b10f
Diffstat (limited to 'tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp')
-rw-r--r-- | tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp | 8 |
1 files changed, 2 insertions, 6 deletions
diff --git a/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp b/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp index f8337a5286..345a0fed98 100644 --- a/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp +++ b/tests/ModelAccuracyTool-Armnn/ModelAccuracyTool-Armnn.cpp @@ -76,7 +76,7 @@ int main(int argc, char* argv[]) "Path to armnn format model file", cxxopts::value<std::string>(modelPath)) ("f,model-format", - "The model format. Supported values: caffe, tensorflow, tflite", + "The model format. Supported values: tensorflow, tflite", cxxopts::value<std::string>(modelFormat)) ("i,input-name", "Identifier of the input tensors in the network separated by comma with no space.", @@ -312,11 +312,7 @@ int main(int argc, char* argv[]) const unsigned int batchSize = 1; // Get normalisation parameters SupportedFrontend modelFrontend; - if (modelFormat == "caffe") - { - modelFrontend = SupportedFrontend::Caffe; - } - else if (modelFormat == "tensorflow") + if (modelFormat == "tensorflow") { modelFrontend = SupportedFrontend::TensorFlow; } |