Альтернатива 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 ответа
- создайте новую виртуальную среду и установите tennsflowjs 0.8.6 с помощью команды: pip install tenorflowjs==0.8.6
- дальнейшую модель можно преобразовать с помощью: 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
- обратитесь: https://github.com/tensorflow/tfjs-converter
Современные версии tensorflowjs
пакет больше не конвертируется из frozen models
, С https://github.com/tensorflow/tfjs-converter:
Примечание. Если вы хотите преобразовать замороженную модель или комплект сеансов TensorFlow, вы можете установить более старые версии пакета pip tenorflowjs, то есть pip install tenorflowjs == 0.8.6.
, Надеюсь, это поможет.