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

Question

我想从这里转换这个Frozen Model（扩展名.pb）：

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

到一个保存的模型 。

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

Answer 1

首先回答，这样做可能但非常混乱，因为在冻结ckpt到pb模型期间，SavedModel所需的所有变量都会转换为常量或其他类型。 在tf中，将“常量”恢复为“变量”有点复杂，例如：

# 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))

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

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