如何使用TensorFlow Frozen_graph util?

时间:2019-03-31 21:12:58

标签: python tensorflow

我训练了小型模型,并使用tf.train.write_graph保存了图形。现在,我有检查点文件和.pbtxt文件。 我尝试使用

创建冻结图
freeze_graph.freeze_graph(input_graph='freeze/Model.pbtxt',
              input_saver="",
              input_binary=False,
              input_checkpoint='freeze/Model.ckpt',
              output_node_names="conv2d/BiasAdd",
              restore_op_name="save/restore_all",
              filename_tensor_name='',
              output_graph='frozenModel.pb',
              clear_devices=True,
                          initializer_nodes=''
              )

但是它将IndexError: list index out of range扔到了initializer_nodes =''行中。

如果我删除此行,则会抛出TypeError: freeze_graph() missing 1 required positional argument: 'initializer_nodes'.

我找不到关于freeze_graph参数的任何有意义的解释。如何解决这个错误?在freeze_graph中,尤其是在initializer_nodes中传递什么值?

0 个答案:

没有答案