如何打印tensorflow 2 saved_model的图层

Question

我使用的是tensorflow 2.6.2，我从tensorflow 2 model zoo下载了该模型我可以使用这个加载模型

import tensorflow as tf
if __name__ == "__main__":
try:
model = tf.saved_model.load("/home/user/git/models_zoo/ssd_mobilenet_v2_320x320_coco17_tpu-8/saved_model/")

但不幸的是，我无法使用下面的看到模型的所有层

for v in model.trainable_variables:
print(v.name)

它应该理想地打印网络中的所有层，但我得到了以下错误

print(model.trainable_variables)
AttributeError: '_UserObject' object has no attribute 'trainable_variables'

有人能告诉我，我在这里做错了什么吗。

Answer 1

我可以使用这个打印

loaded = tf.saved_model.load("/home/user/git/models_zoo/ssd_mobilenet_v2_320x320_coco17_tpu-8/saved_model/")
infer = loaded.signatures["serving_default"]
for v in infer.trainable_variables:
print(v.name)