tensorflow - 在 tensorflowjs_converter 中替代 tf_frozen_model

Question

我是 tensorflow 的新手，并通过 pip 安装了当前的 tensorflowjs(1.1.2)，因为我想将冻结的推理图 ( .pb) 转换为 tensorflowjs 文件 ( --output_format=tensorflowjs)。我遵循了一个--input_format=tf_frozen_model建议的教程，这个版本的 tensorflowjs，但是不再支持这样的选项并返回错误消息error:argument --input_format: invalid choice: 'tf_frozen_model'。此选项是否已完全替换，还是仅重命名？我尝试安装 version tensorflowjs v.0.8.0，这导致了一堆 ImportErrors (for Keras) 和依赖冲突。我在这里想念什么吗？

Answer 1

1. 创建一个新的虚拟环境并使用命令安装 tensorflowjs 0.8.6： pip install tensorflowjs==0.8.6
2. 可以使用以下方法转换进一步的模型：tensorflowjs_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

Answer 2

现代版本的tensorflowjs包不再从frozen models. 来自https://github.com/tensorflow/tfjs-converter：

注意：如果要转换 TensorFlow 冻结模型或会话捆绑包，可以安装旧版本的 tensorflowjs pip 包，即 pip install tensorflowjs==0.8.6

. 希望能帮助到你。