如何知道标签名称是tensorflow保存模型？

Question

如何在tensorflow中知道标记名称是“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

希望有所帮助。