【问题标题】:is there a nice output of Keras model.summary( )?Keras model.summary() 是否有很好的输出?
【发布时间】:2019-05-29 12:34:24
【问题描述】:

是否有可能有一个不错的 keras model.summary() 输出,可以包含在纸上,或者可以绘制在像这样的漂亮表格中。

【问题讨论】:

标签: python tensorflow keras deep-learning


【解决方案1】:

您需要安装 graphvis 和 pydot,但您可能会喜欢这样的结果。它没有制作表格,但我认为图表要好得多。

    from keras.utils import plot_model
    plot_model(model, to_file='model.png', show_shapes=True,show_layer_names=True)

但是,如果您想将多个层嵌套在一起,则必须创建正确命名的子模型。

【讨论】:

  • 感谢您的回复,我之前尝试过这种方法,但是对于深度网络,比如说 Resnet50,它会给您一个很长的图像(因为深度),这就是为什么我在考虑更多表格的原因,所以我可以压缩表示,然后引用我想在我的论文中指定的层(例如层 conv2_branch1 )
  • 你可以将多个图层组合成一个子模型,这会缩短图像,但我同意,这并不理想
猜你喜欢
  • 1970-01-01
  • 1970-01-01
  • 2019-06-25
  • 1970-01-01
  • 2018-04-24
  • 2020-03-15
  • 2021-04-28
  • 2017-05-30
相关资源
最近更新 更多