python-3.x - Keras:在 Jupyter Notebook 中显示模型形状

Question

我有以下代码用于查看我的网络架构。



但是，我也想看到每一层的形状，所以我尝试使用以下内容:

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

输出文件“model.png”看起来不错。但我无法让它显示在 Jupyter Notebook 中。知道我错过了什么吗？谢谢!

Answer 1

由于生成的图像不再是 svg 文件，因此您应该替换 SVG与 Image用

from IPython.display import Image 
... 

plot_model(model, show_shapes=True, show_layer_names=True, to_file='model.png')
Image('model.png')