diff options
Diffstat (limited to 'tests/ImageTensorGenerator')
| -rw-r--r-- | tests/ImageTensorGenerator/ImageTensorGenerator.cpp | 8 | ||||
| -rw-r--r-- | tests/ImageTensorGenerator/README.md | 2 |
2 files changed, 3 insertions, 7 deletions
diff --git a/tests/ImageTensorGenerator/ImageTensorGenerator.cpp b/tests/ImageTensorGenerator/ImageTensorGenerator.cpp index 34dbe1e352..5417eb4055 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 (tensorflow, tflite)", + "Accepted value (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 == "tensorflow") - { - return SupportedFrontend::TensorFlow; - } - else if (m_ModelFormat == "tflite") + if (m_ModelFormat == "tflite") { return SupportedFrontend::TFLite; } diff --git a/tests/ImageTensorGenerator/README.md b/tests/ImageTensorGenerator/README.md index b6bc3126b4..312fcddc5d 100644 --- a/tests/ImageTensorGenerator/README.md +++ b/tests/ImageTensorGenerator/README.md @@ -9,7 +9,7 @@ To build ModelAccuracyTool, pass the following options to Cmake: |Cmd:||| | ---|---|---| | -h | --help | Display help messages | -| -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) | +| -f | --model-format | Format of the intended model file that uses the images.Different formats have different image normalization styles.Accepted values (tflite) | | -i | --infile | Input image file to generate tensor from | | -o | --outfile | Output raw tensor file path | | -z | --output-type | The data type of the output tensors.If unset, defaults to "float" for all defined inputs. Accepted values (float, int or qasymm8) |