我正在尝试从此处转换此卡住模型(扩展名 .pb):
https://www.dropbox.com/s/bb7n78hq5owyjr0/tf_model_pb.zip?dl=0
到Saved Model .
这可能吗?您能否分享将此 pb 文件转换为已保存模型的确切代码?
最佳答案
先回答一下,这样做是可以的,但是很麻烦,因为在卡住ckpt到pb模型的过程中,SavedModel所需的所有变量都被转换为常量或其他类型。在 tf 中,将 'Constant' 恢复为 '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 是最简单、最直接的方法。
关于python - Tensorflow:如何将卡住模型转换为保存模型,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/50394633/