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

Question

如何vgg16在tensorflow中知道标签名称是“ ”。有什么方法可以从保存的模型图/变量转储中取回它？

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

Answer 1

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

例如：

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

希望能有所帮助。