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

Я новичок в tenorflow и установил текущий tenorsflowjs(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) и конфликтам зависимостей. Я что-то здесь скучаю?

2 ответа

  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
  3. обратитесь: https://github.com/tensorflow/tfjs-converter

Современные версии tensorflowjs пакет больше не конвертируется из frozen models, С https://github.com/tensorflow/tfjs-converter:

Примечание. Если вы хотите преобразовать замороженную модель или комплект сеансов TensorFlow, вы можете установить более старые версии пакета pip tenorflowjs, то есть pip install tenorflowjs == 0.8.6.

, Надеюсь, это поможет.

Другие вопросы по тегам