Tensorflow：如何将冻结的模型转换为保存的模型

Question

我正在尝试从这里转换这个 Frozen 模型（扩展名 .pb）：

https://www.dropbox.com/s/bb7n78hq5owyjr0/tf_model_pb.zip?dl=0

到Saved Model。

有可能吗？你能分享一下将这个 pb 文件转换为保存模型的确切代码吗？

Answer 1

首先回答，这样做是可能的，但这样做很麻烦，因为在将 ckpt 冻结为 pb 模型期间，SavedModel 所需的所有变量都会转换为常量或其他类型。在 tf 中，revert 'Constant' to 'Variable' 有点复杂，比如：

# node is a constant
node = tf.Constant([1, 2])
output_node = tf.NodeDef()
output_node.op = "Variable"
output_node.name = node.name
dtype = node.attr["dtype"].type
data = node.attr["value"].tensor
output_node.attr["dtype"].type = dtype
output_node.attr["value"].CopyFrom(tf.AttrValue(tensor=data))

有关将常量节点转换为其他类型的详细信息，请参阅此answer。

因此，从 ckpt 文件中冻结 SavedModel 是最简单、最直接的方法。