Tensorflow: NotImplementedError: В настоящее время могут быть импортированы только SavedModels, экспортированные с `tf.saved_model.save` - PullRequest
Новая
       11

Tensorflow: NotImplementedError: В настоящее время могут быть импортированы только SavedModels, экспортированные с `tf.saved_model.save`

1 голос
/ 27 апреля 2019

У меня есть график тензорного потока, который я сохранил как «save_model.pb», используя этот код: 

builder = tf.saved_model.builder.SavedModelBuilder('saved_location') 
with tf.Session(graph=tf.Graph()) as sess, open('logfile', 'wb') as log_file:
...
...
builder.add_meta_graph_and_variables(sess, [tf.saved_model.tag_constants.SERVING])

Я периодически вызываю builder.save (). и посмотрите, как обновляется файл сохраненный_модель.pb.

теперь, когда я пытаюсь импортировать его для JavaScript, ссылка 

>  tensorflowjs_converter --input_format=tf_saved_model 
 --output_format=tfjs_graph_model saved_location/  target

я получаю следующую ошибку:

Файл "/Anaconda3/lib/python3.5/site-packages/tensorflowjs/converters/tf_saved_model_conversion_v2.py", строка 269, в convert_tf_saved_model Модель = загрузить (сохраненный_модель_директора) Файл "/anaconda3/lib/python3.5/site-packages/tensorflow/python/saved_model/load.py",> строка 108, в загрузке

"В настоящее время только SavedModels, экспортированные с tf.saved_model.save, могут иметь значение" NotImplementedError: В настоящее время только SavedModels экспортируются с tf.saved_model.save может быть импортировано. Другие SavedModels могут в конечном итоге будет поддерживаться через load ().

  1. Где я иду не так?

  2. как я могу преобразовать модель графа тензорного потока в формат, который может быть загружен в браузеры для вывода с использованием javascript?

Примечание: я могу импортировать модель слоев keras в js и использовать ее через API loadLayersModel, но я ищу решение, в котором мне не нужно преобразовывать всю кодовую базу в keras.

...