如何获得TensorFlow模型的总结？

Question

我对这方面还比较新，开始按照这个教程进行目标检测。我已经在最新版本中运行了所有内容，训练并导出了模型，并且正在进行目标检测。

但是我仍然不知道modelzoo中的那些模型从内部看起来是什么样子的。我看过一些模型的摘要，例如这个或者那个问题，但我不知道该在哪里应用它们？

我尝试使用以下命令打开我训练和导出的模型：

model = tf.saved_model.load(MODEL_PATH)

仅需添加一个

model.summary()

我尝试执行该命令，但是它只是报错了

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

那么，我错过了什么？我是不是需要以某种方式加载已训练而不是已导出的模型？如果是，我该怎么做？

- Puki

---

这回答了你的问题吗？在Tensorflow中是否有像Keras model.summary这样的简单方法？ - Trenton McKinney

看起来很有前途，但我无法运行 inspect_pb.py 的简易版本，它只会给我一个 google.protobuf.message.DecodeError: Error parsing message 错误 - 看起来它是为 TF1 设计的，但我正在运行 TF2。 - Puki

Answer 1

尝试：

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

- NicolasPeruchot

---

我尝试了一下，但是又出现了另一个错误：python3.9/site-packages/tensorflow/python/util/dispatch.py"，第206行，在包装器中， 返回目标(*args, **kwargs) TypeError: 'str'对象不可调用 - Puki