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 use

from IPython.display import Image 
... 

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