tensorflow - 在 Tensorflow 2 中导出卡住图 .pb 文件

Question

我一直在尝试 Tensorflow 2 alpha 并且我一直在尝试卡住模型并将其导出到 .pb graphdef 文件。

在 Tensorflow 1 中，我可以做这样的事情:

# Freeze the graph.
frozen_graph_def = tf.graph_util.convert_variables_to_constants(
    sess,
    sess.graph_def,
    output_node_names)

# Save the frozen graph to .pb file.
with open('model.pb', 'wb') as f:
    f.write(frozen_graph_def.SerializeToString())

然而，这似乎不再可能，因为 convert_variables_to_constants 已被删除，并且不鼓励使用 session 。

我查看并发现有卡住图实用程序
https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/freeze_graph.py适用于 SavedModel 导出。

有什么方法可以在 Python 中做到这一点，或者我现在打算切换和使用这个工具？

Answer 1

在从 tensorflow1.x 迁移到 tensoflow2.0 beta 时，我也遇到了同样的问题。
这个问题可以通过2种方法解决:

第一个是去tensflow2.0 docs搜索你使用过的方法并更改每一行的语法&

使用 google 的 tf_ugrade_v2 脚本

tf_upgrade_v2 --infile your_tf1_script_file --outfile Converted_tf2_file

您尝试使用上述命令将 tensorflow1.x 脚本更改为 tensorflow2.0，它将解决您的所有问题。

此外，您可以重命名该方法(通过引用文档手动步骤)
重命名 'tf.graph_util.convert_variables_to_constants' 到 'tf.compat.v1.graph_util.convert_variables_to_constants'

测量问题是，在 tensorflow2.0 中，许多语法和功能发生了变化，请尝试引用 tensoflow2.0 文档或使用谷歌的 tf_upgrade_v2 脚本