[tflite_run] Do not support image inputs (#4098)
author이한종/On-Device Lab(SR)/Engineer/삼성전자 <hanjoung.lee@samsung.com>
Tue, 18 Dec 2018 10:44:43 +0000 (19:44 +0900)
committer이춘석/On-Device Lab(SR)/Staff Engineer/삼성전자 <chunseok.lee@samsung.com>
Tue, 18 Dec 2018 10:44:43 +0000 (19:44 +0900)
The image file as input is not our way of testing any longer. Since this
only support inception v3 IFM, it would be nicer not to have this
option.

Resolve #1206

Signed-off-by: Hanjoung Lee <hanjoung.lee@samsung.com>
tests/tools/tflite_run/src/args.cc
tests/tools/tflite_run/src/args.h
tests/tools/tflite_run/src/tflite_run.cc

index 6aebbbb..713a0a9 100644 (file)
@@ -38,7 +38,6 @@ void Args::Initialize(void)
   // clang-format off
   general.add_options()
     ("help,h", "Display available options")
-    ("input,i", po::value<std::string>(&_input_filename)->default_value(""), "Input filename")
     ("dump,d", po::value<std::string>()->default_value(""), "Output filename")
     ("compare,c", po::value<std::string>()->default_value(""), "filename to be compared with")
     ("tflite", po::value<std::string>()->required());
@@ -55,6 +54,7 @@ void Args::Parse(const int argc, char **argv)
             vm);
   po::notify(vm);
 
+#if 0 // Enable this when we have mutually conflicting options
   {
     auto conflicting_options = [&](const std::string &o1, const std::string &o2) {
       if ((vm.count(o1) && !vm[o1].defaulted()) && (vm.count(o2) && !vm[o2].defaulted()))
@@ -66,6 +66,7 @@ void Args::Parse(const int argc, char **argv)
 
     conflicting_options("input", "compare");
   }
+#endif
 
   if (vm.count("help"))
   {
@@ -77,19 +78,6 @@ void Args::Parse(const int argc, char **argv)
     exit(0);
   }
 
-  if (vm.count("input"))
-  {
-    _input_filename = vm["input"].as<std::string>();
-
-    if (!_input_filename.empty())
-    {
-      if (!boost::filesystem::exists(_input_filename))
-      {
-        std::cerr << "input image file not found: " << _input_filename << "\n";
-      }
-    }
-  }
-
   if (vm.count("dump"))
   {
     _dump_filename = vm["dump"].as<std::string>();
index 7b270d4..5561544 100644 (file)
@@ -32,7 +32,6 @@ public:
   void print(void);
 
   const std::string &getTFLiteFilename(void) const { return _tflite_filename; }
-  const std::string &getInputFilename(void) const { return _input_filename; }
   const std::string &getDumpFilename(void) const { return _dump_filename; }
   const std::string &getCompareFilename(void) const { return _compare_filename; }
 
@@ -45,7 +44,6 @@ private:
   po::options_description _options;
 
   std::string _tflite_filename;
-  std::string _input_filename;
   std::string _dump_filename;
   std::string _compare_filename;
 };
index d0d6924..5be6909 100644 (file)
@@ -90,19 +90,8 @@ int main(const int argc, char **argv)
 
   TFLiteRun::TensorLoader tensor_loader(*interpreter);
 
-  // Load input from image or dumped tensor file. Two options are exclusive and will be checked
-  // from Args.
-  if (args.getInputFilename().size() > 0)
-  {
-    BinImage image(299, 299, 3);
-    image.loadImage(args.getInputFilename());
-
-    for (const auto &o : interpreter->inputs())
-    {
-      image.AssignTensor(interpreter->tensor(o));
-    }
-  }
-  else if (!args.getCompareFilename().empty())
+  // Load input from dumped tensor file.
+  if (!args.getCompareFilename().empty())
   {
     tensor_loader.load(args.getCompareFilename());