如何知道标签名称是张量流保存模型

Question

如何知道标签名称在张量流中是"vgg16"。有没有办法从保存的模型图/变量转储中取回它？

tf.saved_model.loader.load(sess, [“vgg16”], “./data/vgg/”)

Answer 1

您可以使用saved_model_cli命令行工具将存储在模型中的所有元数据转储到磁盘上。当您pip install该工具时，该工具包含在最新版本的TensorFlow(肯定为1.4+(中。

例如：

saved_model_cli show --dir ./data/vgg --all

希望有帮助。