diff options
Diffstat (limited to 'tests/ImageTensorGenerator/ImageTensorGenerator.cpp')
-rw-r--r-- | tests/ImageTensorGenerator/ImageTensorGenerator.cpp | 8 |
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; } |