TFHub 模型的参数数量

Question

我想计算从 TensorFlow Hub 加载的对象检测模型的参数数量，例如https://tfhub.dev/tensorflow/ssd_mobilenet_v2/2 。 我试过这个：

hub_model = hub.load("https://tfhub.dev/tensorflow/ssd_mobilenet_v2/2")
print(len(hub_model.signatures['serving_default'].variables))

但是输出不是很可读，我什至不确定它是否正确。 我也试过这种方式：

malli = hub.KerasLayer("https://tfhub.dev/tensorflow/ssd_mobilenet_v2/2")
print("Thickness of the model:", len(malli.weights))

但它只返回一个长度为 0 的空列表 []。

能够在这些模型上使用 Keras summary() 方法会很好，但不能在 KerasLayer 上调用它，那么将此层合并到具有 Keras.Sequential 的模型中会起作用吗？

Answer 1

有一个工具可以计算检查点文件的参数总数： https : //github.com/tensorflow/tensorflow/blob/master/tensorflow/python/tools/inspect_checkpoint.py

$ python inspect_checkpoint.py --file_name=/checkpoint/file/name --all_tensors