tensorflowjs_converter中的tf_frozen_model的替代品

时间:2019-06-12 20:45:27

标签: tensorflow tensorflowjs-converter tensorflowjs

我是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'。此选项是否已完全替换或仅重命名? 我试图安装版本tensorflowjs v.0.8.0,这导致了一堆ImportErrors(对于Keras)和依赖冲突。我在这里想念什么吗?

2 个答案:

答案 0 :(得分:0)

  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. 引用:{{3}}

答案 1 :(得分:0)

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

  

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

。 希望对您有所帮助。