以图形方式显示在 Keras 中编码的模型

Question

假设我在 keras 中从头开始编写 Resnet 152 模型，是否存在任何 tool/api 可以将我的代码作为输入并给出图形信号流图，以便我可以确定我编写的模型符合预期吗？

我知道存在 Keras 模型摘要，但我如何从中解释跳过层连接？

Answer 1

试试看 plot_model(your_model, to_file='model.png')

您将获得模型的 .png 文件，其中显示了层的连接方式。