aboutsummaryrefslogtreecommitdiff
path: root/tests/ImageTensorGenerator/ImageTensorGenerator.cpp
diff options
context:
space:
mode:
authorNikhil Raj <nikhil.raj@arm.com>2021-04-02 22:04:39 +0100
committerNikhil Raj <nikhil.raj@arm.com>2021-04-16 14:27:27 +0100
commit6dd178f2395b34cfb360eabb0130c19ed258f5fa (patch)
tree6814e4729bbd1e652d8b9c18c9db28f3bc2f8a8a /tests/ImageTensorGenerator/ImageTensorGenerator.cpp
parented7fce413410d15c501ea52f9e6bfbbf71b3daf1 (diff)
downloadarmnn-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/ImageTensorGenerator/ImageTensorGenerator.cpp')
-rw-r--r--tests/ImageTensorGenerator/ImageTensorGenerator.cpp8
1 files changed, 2 insertions, 6 deletions
diff --git a/tests/ImageTensorGenerator/ImageTensorGenerator.cpp b/tests/ImageTensorGenerator/ImageTensorGenerator.cpp
index 754d980423..34dbe1e352 100644
--- a/tests/ImageTensorGenerator/ImageTensorGenerator.cpp
+++ b/tests/ImageTensorGenerator/ImageTensorGenerator.cpp
@@ -164,7 +164,7 @@ public:
("f,model-format",
"Format of the intended model file that uses the images."
"Different formats have different image normalization styles."
- "Accepted values (caffe, tensorflow, tflite)",
+ "Accepted values (tensorflow, tflite)",
cxxopts::value<std::string>(m_ModelFormat))
("o,outfile",
"Output raw tensor file path",
@@ -235,11 +235,7 @@ public:
unsigned int GetNewHeight() {return static_cast<unsigned int>(std::stoi(m_NewHeight));}
SupportedFrontend GetModelFormat()
{
- if (m_ModelFormat == "caffe")
- {
- return SupportedFrontend::Caffe;
- }
- else if (m_ModelFormat == "tensorflow")
+ if (m_ModelFormat == "tensorflow")
{
return SupportedFrontend::TensorFlow;
}