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 /src/armnnConverter/ArmnnConverter.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 'src/armnnConverter/ArmnnConverter.cpp')
-rw-r--r-- | src/armnnConverter/ArmnnConverter.cpp | 26 |
1 files changed, 3 insertions, 23 deletions
diff --git a/src/armnnConverter/ArmnnConverter.cpp b/src/armnnConverter/ArmnnConverter.cpp index b5d3256c3c..6f05325bba 100644 --- a/src/armnnConverter/ArmnnConverter.cpp +++ b/src/armnnConverter/ArmnnConverter.cpp @@ -4,9 +4,6 @@ // #include <armnn/Logging.hpp> -#if defined(ARMNN_CAFFE_PARSER) -#include <armnnCaffeParser/ICaffeParser.hpp> -#endif #if defined(ARMNN_ONNX_PARSER) #include <armnnOnnxParser/IOnnxParser.hpp> #endif @@ -81,9 +78,6 @@ int ParseCommandLineArgs(int argc, char* argv[], try { std::string modelFormatDescription("Format of the model file"); -#if defined(ARMNN_CAFFE_PARSER) - modelFormatDescription += ", caffe-binary, caffe-text"; -#endif #if defined(ARMNN_ONNX_PARSER) modelFormatDescription += ", onnx-binary, onnx-text"; #endif @@ -336,11 +330,10 @@ private: int main(int argc, char* argv[]) { -#if (!defined(ARMNN_CAFFE_PARSER) \ - && !defined(ARMNN_ONNX_PARSER) \ +#if (!defined(ARMNN_ONNX_PARSER) \ && !defined(ARMNN_TF_PARSER) \ && !defined(ARMNN_TF_LITE_PARSER)) - ARMNN_LOG(fatal) << "Not built with any of the supported parsers, Caffe, Onnx, Tensorflow, or TfLite."; + ARMNN_LOG(fatal) << "Not built with any of the supported parsers Onnx, Tensorflow, or TfLite."; return EXIT_FAILURE; #endif @@ -399,20 +392,7 @@ int main(int argc, char* argv[]) try { - if (modelFormat.find("caffe") != std::string::npos) - { -#if defined(ARMNN_CAFFE_PARSER) - if (!converter.CreateNetwork<armnnCaffeParser::ICaffeParser>()) - { - ARMNN_LOG(fatal) << "Failed to load model from file"; - return EXIT_FAILURE; - } -#else - ARMNN_LOG(fatal) << "Not built with Caffe parser support."; - return EXIT_FAILURE; -#endif - } - else if (modelFormat.find("onnx") != std::string::npos) + if (modelFormat.find("onnx") != std::string::npos) { #if defined(ARMNN_ONNX_PARSER) if (!converter.CreateNetwork<armnnOnnxParser::IOnnxParser>()) |