альтернатива tf_frozen_model в tenorflowjs_converter - PullRequest
Новая
       51

альтернатива tf_frozen_model в tenorflowjs_converter

0 голосов
/ 12 июня 2019

Я новичок в tenorflow и установил текущий tenorflowjs (1.1.2) через pip, поскольку хочу преобразовать замороженный граф вывода (.pb) в файл tenorflowjs (--output_format=tensorflowjs).Я следовал руководству, где предлагается --input_format=tf_frozen_model, эта версия tenorflowjs, однако больше не поддерживает такую ​​опцию и возвращает сообщение об ошибке error:argument --input_format: invalid choice: 'tf_frozen_model'.Эта опция была полностью заменена или только переименована?Я попытался установить версию tensorflowjs v.0.8.0, что привело к куче ошибок ImportErrors (для Keras) и зависимостей.Я что-то здесь скучаю?

1 Ответ

0 голосов
/ 02 июля 2019
  1. Создайте новую виртуальную среду и установите tennsflowjs 0.8.6 с помощью команды: pip install tenorflowjs == 0.8.6
  2. дальнейшую модель можно преобразовать с помощью: tenorsflowjs_converter --input_format = tf_frozen_model --output_json = true --output_node_names = 'MobilenetV1 / Predictions / Reshape_1' --saved_model_tags = serve \ /mobilenet/frozen_model.pb / mobilenet / web_model
  3. см https://github.com/tensorflow/tfjs-converter
...