tenorflowjs_converter: файл SavedModel не существует в:

Question

Я следую инструкциям здесь , чтобы импортировать мою модель Tensorflow, созданную с помощью Python, в Tensorflow.js.

Модель создается с использованием tf.estimator API , который автоматически сохраняет файлы модели в соответствии с этой страницей .

После установки tensorflowjs с помощью pip, я получаю следующую ошибку при запуске скрипта конвертации:

$ tensorflowjs_converter     --input_format=tf_saved_model     --output_node_names='DeepWater'     /home/sterengiannini/deepwater/model-js-import-test     /home/sterengiannini/deepwater/web-model
Using TensorFlow backend.
Traceback (most recent call last):
  File "/usr/local/bin/tensorflowjs_converter", line 11, in <module>
    sys.exit(main())
  File "/home/sterengiannini/.local/lib/python2.7/site-packages/tensorflowjs/converters/converter.py", line 317, in main
    strip_debug_ops=FLAGS.strip_debug_ops)
  File "/home/sterengiannini/.local/lib/python2.7/site-packages/tensorflowjs/converters/tf_saved_model_conversion.py", line 289, in convert_tf_saved_model
    input_saved_model_dir=saved_model_dir)
  File "/home/sterengiannini/.local/lib/python2.7/site-packages/tensorflow/python/tools/freeze_graph.py", line 338, in freeze_graph
    input_saved_model_dir, saved_model_tags).graph_def
  File "/home/sterengiannini/.local/lib/python2.7/site-packages/tensorflow/python/tools/saved_model_utils.py", line 42, in get_meta_graph_def
    saved_model = reader.read_saved_model(saved_model_dir)
  File "/home/sterengiannini/.local/lib/python2.7/site-packages/tensorflow/contrib/saved_model/python/saved_model/reader.py", line 55, in read_saved_model
    raise IOError("SavedModel file does not exist at: %s" % saved_model_dir)
IOError: SavedModel file does not exist at: /home/sterengiannini/deepwater/model-js-import-test

Вот файлы, представленные в папке модели:

$ ls model-js-import-test/
graph.pbtxt  model.ckpt-312.data-00000-of-00001  model.ckpt-312.index  model.ckpt-312.meta

Answer 1

Похоже, модель не была сохранена в нужном формате.Вы должны ожидать что-то вроде следующего в папке модели ( см. Структура каталога SavedModel ):

assets/
assets.extra/
variables/
    variables.data-?????-of-?????
    variables.index
saved_model.pb|saved_model.pbtxt

Например, если вы используете API tf.estimator, вам нужноЧтобы экспортировать модель со следующим ( см. Использование SavedModel с оценщиками ):

estimator.export_savedmodel(export_dir_base, serving_input_receiver_fn)