googlecodelabs / tensorflow-for-poets-2

Apache License 2.0
509 stars 463 forks source link

Optimize for inference - Mobile optimizing graph error. #21

Open MauryaRitesh opened 6 years ago

MauryaRitesh commented 6 years ago

I got the following error, while optimizing for mobile: Traceback (most recent call last): File "/usr/lib/python3.6/runpy.py", line 193, in _run_module_as_main "main", mod_spec) File "/usr/lib/python3.6/runpy.py", line 85, in _run_code exec(code, run_globals) File "/home/riteshk_m_786/.local/lib/python3.6/site-packages/tensorflow/python/tools/optimize_for_inference.py", line 146, in app.run(main=main, argv=[sys.argv[0]] + unparsed) File "/home/riteshk_m_786/.local/lib/python3.6/site-packages/tensorflow/python/platform/app.py", line 48, in run _sys.exit(main(_sys.argv[:1] + flags_passthrough)) File "/home/riteshk_m_786/.local/lib/python3.6/site-packages/tensorflow/python/tools/optimize_for_inference.py", line 90, in main FLAGS.output_names.split(","), FLAGS.placeholder_type_enum) File "/home/riteshk_m_786/.local/lib/python3.6/site-packages/tensorflow/python/tools/optimize_for_inference_lib.py", line 109, in optimize_for_inference placeholder_type_enum) File "/home/riteshk_m_786/.local/lib/python3.6/site-packages/tensorflow/python/tools/strip_unused_lib.py", line 83, in strip_unused raise KeyError("The following input nodes were not found: %s\n" % not_found) KeyError: "The following input nodes were not found: {'input'}\n"

BharathBillawa commented 6 years ago

If you are using mobilenet the input node will be "input", but in inception its "mul" . So just replace that ... I think that'll work.. :)

SagarSharma4244 commented 6 years ago

If you are getting an error KeyError: "The following input nodes were not found: {'input'}\n" then change "input" to "Mul" If anybody still has doubt refer TensorFlow on Mobile: Tutorial