检查PB文件中DL模型的输入和输出尺寸

Question

我使用几种DL模型，通常以TensorFlow PB文件的形式提供，没有其他信息。 我需要从PB文件中提取输入和输出层的名称和尺寸。 该怎么办？

Answer 1

您可以使用 Netron 软件。可以从here下载。它是用于DL模型可视化的工具，如下所示：