如何获取 TensorFlow 模型的摘要？答案

【问题标题】：How to get summary of TensorFlow model?如何获取 TensorFlow 模型的摘要？
【发布时间】：2021-12-16 02:38:55
【问题描述】：

我对这些东西还很陌生，并在this tutorial 之后开始进行对象检测。我的所有东西都在最新版本中运行，我已经训练和导出了模型，并且我正在做对象检测。

但我仍然不知道 modelzoo 中的那些模型在引擎盖下的样子。我在this 或that 问题中看到了模型摘要，但我不知道在哪里应用？

我试图用

打开我训练和导出的模型

model = tf.saved_model.load(MODEL_PATH)

只需添加一个

model.summary()

命令，但它只是给我一个错误提示

AttributeError: '_UserObject' object has no attribute 'summary'

那么，我错过了什么？我是否必须以某种方式加载经过训练的模型而不是导出的模型？如果是，我应该怎么做？

【问题讨论】：

这能回答你的问题吗？ Is there an easy way to get something like Keras model.summary in Tensorflow?
这看起来很有希望，虽然我无法使用 inspect_pb.py 运行简单版本，但它只是给了我一个 google.protobuf.message.DecodeError: Error parsing message - 似乎它是为 TF1 设计的，但我正在运行 TF2跨度>

【解决方案1】：

试试：

model = tf.keras.models.load_model(MODEL_PATH)

【讨论】：

我试过了，只是又报错：python3.9/site-packages/tensorflow/python/util/dispatch.py", line 206, in wrapper return target(*args, **kwargs) TypeError: 'str' object is not callable